[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/litmodels/integrations/checkpoints.py

@@ -1,4 +1,4 @@
-from typing import Any, Type, TypeVar, cast
+from typing import Any, TypeVar
 
 from lightning_sdk.lightning_cloud.login import Auth
 from lightning_utilities.core.rank_zero import rank_zero_only
@@ -38,6 +38,7 @@ def _upload_model(self, filepath: str) -> None:
 
 # Create specific implementations
 if _LIGHTNING_AVAILABLE:
+
     class LightningModelCheckpoint(LitModelCheckpointMixin, _LightningModelCheckpoint):
         """Lightning ModelCheckpoint with LitModel support.
 
@@ -55,7 +56,9 @@ def _save_checkpoint(self, trainer: "Trainer", filepath: str) -> None:
             super()._save_checkpoint(trainer, filepath)
             self._upload_model(filepath)
 
+
 if _PYTORCHLIGHTNING_AVAILABLE:
+
     class PTLightningModelCheckpoint(LitModelCheckpointMixin, _PytorchLightningModelCheckpoint):
         """PyTorch Lightning ModelCheckpoint with LitModel support.
 
@@ -71,4 +74,4 @@ def __init__(self, model_name: str, *args: Any, **kwargs: Any) -> None:
 
         def _save_checkpoint(self, trainer: "Trainer", filepath: str) -> None:
             super()._save_checkpoint(trainer, filepath)
-            self._upload_model(filepath)
+            self._upload_model(filepath)