Update lightning import in utils.py and bump version to 0.9.7 in setu…

…p.py
BardiaKh · Jan 31, 2024 · 6821fe4 · 6821fe4
1 parent 99e705f
commit 6821fe4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bkh_pytorch_utils/pl/utils.py b/bkh_pytorch_utils/pl/utils.py
@@ -5,7 +5,7 @@
 from overrides import overrides
 from tabulate import tabulate
 import lightning as pl
-from lightning.utilities import rank_zero_only
+from lightning.pytorch.utilities import rank_zero_only
 from torch.utils.data.distributed import DistributedSampler
 from .ddp_helper import DistributedProxySampler
 

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="bkh_pytorch_utils",
-    version="0.9.6",
+    version="0.9.7",
     author="Bardia Khosravi",
     author_email="[email protected]",
     description="A rapid prototyping tool for MONAI & PyTorch Lightning",