diff --git a/optax/contrib/reduce_on_plateau_test.py b/optax/contrib/reduce_on_plateau_test.py index 7f77fe29..3edd2b07 100644 --- a/optax/contrib/reduce_on_plateau_test.py +++ b/optax/contrib/reduce_on_plateau_test.py @@ -23,7 +23,7 @@ class ReduceLROnPlateauTest(absltest.TestCase): def test_learning_rate_reduced_after_cooldown_period_is_over(self): - """Test that learning rate is reduced again after cooldown period is over.""" + """Test that learning rate is reduced after the cooldown period.""" # Define a dummy update and extra_args updates = {'params': jnp.array(1.0)}