Use a different step to store nan as an intermediate value

optuna · Feb 22, 2024 · 6f136a0 · 6f136a0
1 parent 89b3058
commit 6f136a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_keras.py b/tests/test_keras.py
@@ -53,7 +53,7 @@ def test_keras_pruning_callback_observation_isnan() -> None:
         callback.on_epoch_end(0, {"loss": 1.0})
 
     with pytest.raises(optuna.TrialPruned):
-        callback.on_epoch_end(0, {"loss": float("nan")})
+        callback.on_epoch_end(1, {"loss": float("nan")})
 
 
 def test_keras_pruning_callback_monitor_is_invalid() -> None: