formatting

Author: Borda

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.