Skip to content

Commit 93de81e

Browse files
committed
Wrong name
1 parent 60de814 commit 93de81e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/trainer/test_trainer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -723,9 +723,9 @@ def predict_step(self, batch, *_):
723723
assert getattr(trainer, path_attr) == ckpt_path
724724

725725

726-
@pytest.mark.parametrize("enable_model_summary", (False, True))
726+
@pytest.mark.parametrize("enable_checkpointing", (False, True))
727727
@pytest.mark.parametrize("fn", ("validate", "test", "predict"))
728-
def test_tested_checkpoint_path_best(tmpdir, enable_model_summary, fn):
728+
def test_tested_checkpoint_path_best(tmpdir, enable_checkpointing, fn):
729729
class TestModel(BoringModel):
730730
def validation_step(self, batch, batch_idx):
731731
self.log("foo", -batch_idx)
@@ -746,15 +746,15 @@ def predict_step(self, batch, *_):
746746
limit_predict_batches=1,
747747
enable_progress_bar=False,
748748
default_root_dir=tmpdir,
749-
enable_model_summary=enable_model_summary,
749+
enable_checkpointing=enable_checkpointing,
750750
)
751751
trainer.fit(model)
752752

753753
trainer_fn = getattr(trainer, fn)
754754
path_attr = f"{fn}{'d' if fn == 'validate' else 'ed'}_ckpt_path"
755755
assert getattr(trainer, path_attr) is None
756756

757-
if enable_model_summary:
757+
if enable_checkpointing:
758758
trainer_fn(ckpt_path="best")
759759
assert getattr(trainer, path_attr) == trainer.checkpoint_callback.best_model_path
760760

0 commit comments

Comments
 (0)