fix tests

SkafteNicki · SkafteNicki · commit 7b80a0e6d24d · 2025-10-20T07:18:20.000+02:00
diff --git a/tests/tests_pytorch/checkpointing/test_model_checkpoint.py b/tests/tests_pytorch/checkpointing/test_model_checkpoint.py
@@ -1948,7 +1948,7 @@ def test_save_last_every_n_epochs_interaction(tmp_path, every_n_epochs):
     with patch.object(trainer, "save_checkpoint") as save_mock:
         trainer.fit(model)
     assert mc.last_model_path  # a "last" ckpt was saved
-    assert save_mock.call_count == trainer.max_epochs
+    assert save_mock.call_count == trainer.max_epochs - 1
 
 
 def test_train_epoch_end_ckpt_with_no_validation():
