fix: modify and sort imports

Author: gonzachiar

src/lightning/pytorch/callbacks/model_checkpoint.py

@@ -35,21 +35,21 @@
 from torch import Tensor
 from typing_extensions import override
 
-import pytorch_lightning as pl
+import lightning.pytorch as pl
+from lightning.pytorch.callbacks import Checkpoint
+from lightning.pytorch.utilities.exceptions import MisconfigurationException
+from lightning.pytorch.utilities.rank_zero import (
+    WarningCache,
+    rank_zero_info,
+    rank_zero_warn,
+)
+from lightning.pytorch.utilities.types import STEP_OUTPUT
 from lightning_fabric.utilities.cloud_io import (
     _is_dir,
     _is_local_file_protocol,
     get_filesystem,
 )
 from lightning_fabric.utilities.types import _PATH
-from pytorch_lightning.callbacks import Checkpoint
-from pytorch_lightning.utilities.exceptions import MisconfigurationException
-from pytorch_lightning.utilities.rank_zero import (
-    WarningCache,
-    rank_zero_info,
-    rank_zero_warn,
-)
-from pytorch_lightning.utilities.types import STEP_OUTPUT
 
 log = logging.getLogger(__name__)
 warning_cache = WarningCache()
@@ -252,7 +252,7 @@ def __init__(
         self.best_k_models: dict[str, dict[str, Tensor | dict[str, Tensor]]] = {}
         self.kth_best_model_path = ""
         self.best_model_score: Optional[Tensor] = None
-        self.best_model_metrics: Optional[Dict[str, Tensor]] = None
+        self.best_model_metrics: Optional[dict[str, Tensor]] = None
         self.best_model_path = ""
         self.last_model_path = ""
         self._last_checkpoint_saved = ""