Update pytorch_lightning/core/hooks.py

Co-authored-by: ananthsub <[email protected]>
Lightning-AI · Aug 27, 2021 · b85c4b6 · b85c4b6
1 parent 2e39178
commit b85c4b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/core/hooks.py b/pytorch_lightning/core/hooks.py
@@ -18,7 +18,8 @@
 import torch
 from torch.optim.optimizer import Optimizer
 
-from pytorch_lightning.utilities import move_data_to_device, rank_zero_deprecation, rank_zero_warn
+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