Update path type annotation for load_from_checkpoint (#15540)

(cherry picked from commit 96c5744)
Lightning-AI · Nov 8, 2022 · ea136f5 · ea136f5
1 parent 4d42e03
commit ea136f5
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
@@ -57,9 +57,9 @@ class ModelIO:
     @classmethod
     def load_from_checkpoint(
         cls,
-        checkpoint_path: Union[str, IO],
+        checkpoint_path: Union[_PATH, IO],
         map_location: _MAP_LOCATION_TYPE = None,
-        hparams_file: Optional[str] = None,
+        hparams_file: Optional[_PATH] = None,
         strict: bool = True,
         **kwargs: Any,
     ) -> Self:  # type: ignore[valid-type]