formatting

Lightning-AI · Aug 7, 2020 · 4af5fdd · 4af5fdd
1 parent c1c96db
commit 4af5fdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.