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

for more information, see https://pre-commit.ci
VectorInstitute · Nov 11, 2024 · 8049694 · 8049694
1 parent dda63c3
commit 8049694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cyclops/models/wrappers/pt_model.py b/cyclops/models/wrappers/pt_model.py
@@ -1309,7 +1309,7 @@ def save_model(self, filepath: str, overwrite: bool = True, **kwargs):
         if include_lr_scheduler:
             state_dict["lr_scheduler"] = self.lr_scheduler_.state_dict()  # type: ignore[attr-defined]
 
-        epoch = kwargs.get("epoch", None)
+        epoch = kwargs.get("epoch")
         if epoch is not None:
             filename, extension = os.path.basename(filepath).split(".")
             filepath = join(