From b512b05d96d2fc35da50fcc171e53de089758945 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Fri, 7 Aug 2020 11:30:01 +0200 Subject: [PATCH] formatting --- pytorch_lightning/trainer/states.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/states.py b/pytorch_lightning/trainer/states.py index 3b51c1282141cd..9e7869d7e48a37 100644 --- a/pytorch_lightning/trainer/states.py +++ b/pytorch_lightning/trainer/states.py @@ -14,7 +14,7 @@ class TrainerState(Enum): INTERRUPTED = 'INTERRUPTED' -def trainer_state(*, entering: Optional[TrainerState] = None, exiting: Optional[TrainerState] = None) -> Callable: +def trainer_state(entering: Optional[TrainerState] = None, exiting: Optional[TrainerState] = None) -> Callable: """ Decorator for :class:`~pytorch_lightning.trainer.trainer.Trainer` methods which changes state to `entering` before the function execution and `exiting` after the function is executed. If `None` is passed to `entering`, the state is not changed.