Skip to content

Commit 2d87b2b

Browse files
[pre-commit.ci] Add auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 7d89837 commit 2d87b2b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cyclops/models/wrappers/pt_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1309,7 +1309,7 @@ def save_model(self, filepath: str, overwrite: bool = True, **kwargs):
13091309
if include_lr_scheduler:
13101310
state_dict["lr_scheduler"] = self.lr_scheduler_.state_dict() # type: ignore[attr-defined]
13111311

1312-
epoch = kwargs.get("epoch", None)
1312+
epoch = kwargs.get("epoch")
13131313
if epoch is not None:
13141314
filename, extension = os.path.basename(filepath).split(".")
13151315
filepath = join(

0 commit comments

Comments
 (0)