Backward compatibility for breaking changes of imports from PL (#965)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Lightning-Universe · May 18, 2023 · 424b93b · 424b93b
1 parent 2e83686
commit 424b93b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pl_bolts/callbacks/data_monitor.py b/pl_bolts/callbacks/data_monitor.py
@@ -14,6 +14,7 @@
 from pl_bolts.utils.stability import under_review
 from pl_bolts.utils.warnings import warn_missing_pkg
 
+# Backward compatibility for Lightning Logger
 try:
  from pytorch_lightning.loggers import Logger
 except ImportError:

diff --git a/pl_bolts/models/detection/retinanet/retinanet_module.py b/pl_bolts/models/detection/retinanet/retinanet_module.py
@@ -135,7 +135,11 @@ def configure_optimizers(self):
 
 @under_review()
 def cli_main():
- from pytorch_lightning.utilities.cli import LightningCLI
+ # Backward compatibility for Lightning CLI
+ try:
+ from pytorch_lightning.utilities.cli import LightningCLI
+ except ImportError:
+ from pytorch_lightning.cli import LightningCLI
 
  from pl_bolts.datamodules import VOCDetectionDataModule