revert

Lightning-AI · Nov 3, 2022 · 9a1e2ee · 9a1e2ee
1 parent 1534747
commit 9a1e2ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/pytorch_lightning/core/saving.py b/src/pytorch_lightning/core/saving.py
@@ -49,8 +49,8 @@
 # the older shall be on the top
 CHECKPOINT_PAST_HPARAMS_KEYS = ("hparams", "module_arguments")  # used in 0.7.6
 
-LM = TypeVar("LM", bound=pl.LightningModule)
-LDM = TypeVar("LDM", bound=pl.LightningDataModule)
+LM = TypeVar("LM", bound="pl.LightningModule")
+LDM = TypeVar("LDM", bound="pl.LightningDataModule")
 
 
 class ModelIO: