diff --git a/torchbearer/callbacks/decorators.py b/torchbearer/callbacks/decorators.py index 47bbc0d9..7c4d9f8d 100644 --- a/torchbearer/callbacks/decorators.py +++ b/torchbearer/callbacks/decorators.py @@ -297,9 +297,9 @@ def ope(state): def only_if(condition_expr): """ Decorator to fire a callback only if the given conditional expression function returns True. - :param condition_expr: a function/lambda that must evaluate to true for the - decorated `torchbearer callback` to be called. The `state` - object passed to the callback will be passed as an argument + :param condition_expr: a function/lambda that must evaluate to true for the\ + decorated `torchbearer callback` to be called. The `state`\ + object passed to the callback will be passed as an argument\ to the condition function. :return: the decorator """