diff --git a/pytorch_lightning/core/hooks.py b/pytorch_lightning/core/hooks.py index 6565cad24f248..7361f6d55bdc3 100644 --- a/pytorch_lightning/core/hooks.py +++ b/pytorch_lightning/core/hooks.py @@ -19,8 +19,8 @@ from torch.optim.optimizer import Optimizer from pytorch_lightning.utilities import move_data_to_device -from pytorch_lightning.utilities.warnings import rank_zero_deprecation, rank_zero_warn from pytorch_lightning.utilities.types import EVAL_DATALOADERS, STEP_OUTPUT, TRAIN_DATALOADERS +from pytorch_lightning.utilities.warnings import rank_zero_deprecation, rank_zero_warn class ModelHooks: