From 2d87b2bcfcf0294cbf478ceeafd87548c173aa69 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 4 Nov 2024 21:38:26 +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(