disable trainer prog bar for test of model checkpoint on exception

Author: vsey

tests/tests_pytorch/checkpointing/test_model_checkpoint.py

@@ -773,7 +773,7 @@ def training_step(self, batch, batch_idx):
 
     model = TroubledModel()
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     print(os.listdir(tmp_path))
@@ -789,7 +789,7 @@ def validation_step(self, batch, batch_idx):
     model = TroubledModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -804,7 +804,7 @@ def on_train_batch_start(self, trainer, pl_module, batch, batch_idx):
 
     model = BoringModel()
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainBatchStart()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainBatchStart()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / "step=1.ckpt")
@@ -819,7 +819,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx):
 
     model = BoringModel()
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainBatchEnd()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainBatchEnd()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
 
@@ -836,7 +836,7 @@ def on_train_epoch_start(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainEpochStart()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainEpochStart()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -852,7 +852,7 @@ def on_train_epoch_end(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainEpochEnd()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnTrainEpochEnd()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={2*epoch_length}.ckpt")
@@ -868,7 +868,7 @@ def on_validation_batch_start(self, trainer, pl_module, batch, batch_idx):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationBatchStart()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationBatchStart()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -884,7 +884,7 @@ def on_validation_batch_end(self, trainer, pl_module, outputs, batch, batch_idx)
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationBatchEnd()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationBatchEnd()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -900,7 +900,7 @@ def on_validation_epoch_start(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEpochStart()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEpochStart()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -916,7 +916,7 @@ def on_validation_epoch_end(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEpochEnd()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEpochEnd()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -931,7 +931,7 @@ def on_validation_start(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationStart()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationStart()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")
@@ -946,7 +946,7 @@ def on_validation_end(self, trainer, pl_module):
     model = BoringModel()
     epoch_length = 64
     checkpoint_callback = ModelCheckpoint(dirpath=tmp_path, filename="{step}", save_on_exception=True, every_n_epochs=4)
-    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEnd()], max_epochs=5, logger=False)
+    trainer = Trainer(default_root_dir=tmp_path, callbacks=[checkpoint_callback, TroublemakerOnValidationEnd()], max_epochs=5, logger=False, enable_progress_bar=False)
     with pytest.raises(RuntimeError, match="Trouble!"):
         trainer.fit(model)
     assert os.path.isfile(tmp_path / f"step={epoch_length}.ckpt")