From 55d4f2c69855f9286a4a41d353b769161360dd00 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
<66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 28 Oct 2024 21:44:08 +0000
Subject: [PATCH] [pre-commit.ci] Add auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
---
cyclops/models/wrappers/pt_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/cyclops/models/wrappers/pt_model.py b/cyclops/models/wrappers/pt_model.py
index 50b3a4302..1c54b948c 100644
--- 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(