Merge branch 'bugfix/early-stopping-state--org' into bugfix/early-sto…

…pping-state
Lightning-AI · Jun 28, 2020 · 4cabb88 · 4cabb88
2 parents ae75fa4 + ba6a5ba
commit 4cabb88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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