Skip to content

Commit 4cabb88

Browse files
committed
Merge branch 'bugfix/early-stopping-state--org' into bugfix/early-stopping-state
2 parents ae75fa4 + ba6a5ba commit 4cabb88

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

tests/loggers/test_all.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ def log_metrics(self, metrics, step):
4949
logger = StoreHistoryLogger(**logger_args)
5050

5151
trainer = Trainer(
52+
default_root_dir=tmpdir,
5253
max_epochs=1,
5354
logger=logger,
5455
limit_train_batches=0.2,

tests/trainer/test_trainer_steps.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,11 @@ def test_train_step_epoch_end(tmpdir):
117117
model.training_epoch_end = model.training_epoch_end_dict
118118
model.val_dataloader = None
119119

120-
trainer = Trainer(max_epochs=1, weights_summary=None)
120+
trainer = Trainer(
121+
default_root_dir=tmpdir,
122+
max_epochs=1,
123+
weights_summary=None,
124+
)
121125
trainer.fit(model)
122126

123127
# make sure correct steps were called

0 commit comments

Comments
 (0)