diff --git a/tests/loggers/test_all.py b/tests/loggers/test_all.py index ca309f42afeeed..d66eace4434c57 100644 --- a/tests/loggers/test_all.py +++ b/tests/loggers/test_all.py @@ -49,6 +49,7 @@ def log_metrics(self, metrics, step): logger = StoreHistoryLogger(**logger_args) trainer = Trainer( + default_root_dir=tmpdir, max_epochs=1, logger=logger, limit_train_batches=0.2, diff --git a/tests/trainer/test_trainer_steps.py b/tests/trainer/test_trainer_steps.py index a5ca3c7ab916af..32b17139c15f4a 100644 --- a/tests/trainer/test_trainer_steps.py +++ b/tests/trainer/test_trainer_steps.py @@ -117,7 +117,11 @@ def test_train_step_epoch_end(tmpdir): model.training_epoch_end = model.training_epoch_end_dict model.val_dataloader = None - trainer = Trainer(max_epochs=1, weights_summary=None) + trainer = Trainer( + default_root_dir=tmpdir, + max_epochs=1, + weights_summary=None, + ) trainer.fit(model) # make sure correct steps were called