Skip to content

Commit

Permalink
Update test
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed Feb 24, 2021
1 parent 1fb4e92 commit 3fa2d05
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/callbacks/test_early_stopping.py
Original file line number Diff line number Diff line change
Expand Up @@ -334,13 +334,13 @@ def validation_epoch_end(self, outputs):
# Compute min_epochs latest step
by_min_epochs = min_epochs * limit_train_batches

# Make sure the trainer stops for the max of all minimun requirements
# Make sure the trainer stops for the max of all minimum requirements
assert trainer.global_step == max(min_steps, by_early_stopping, by_min_epochs), \
(trainer.global_step, max(min_steps, by_early_stopping, by_min_epochs), step_freeze, min_steps, min_epochs)

_logger.disabled = False


def test_early_stopping_mode_options():
with pytest.raises(MisconfigurationException, match="`mode` can be auto, .* got unknown_option"):
with pytest.raises(MisconfigurationException, match="`mode` can be .* got unknown_option"):
EarlyStopping(mode="unknown_option")

0 comments on commit 3fa2d05

Please sign in to comment.