Change "pytorch_lightning" to "lightning" in instantiators.py (#577)

ashleve · May 12, 2023 · ca07865 · ca07865
1 parent 467a974
commit ca07865
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils/instantiators.py b/src/utils/instantiators.py
@@ -1,9 +1,9 @@
 from typing import List
 
 import hydra
+from lightning import Callback
+from lightning.pytorch.loggers import Logger
 from omegaconf import DictConfig
-from pytorch_lightning import Callback
-from pytorch_lightning.loggers import Logger
 
 from src.utils import pylogger