Merge pull request #94 from nzw0301/remove-warning-by-calling-trial.r…

…eport-at-the-same-step Remove `UserWarning` by `tests/test_keras.py`
optuna · Feb 26, 2024 · ebc0c25 · ebc0c25
2 parents 89b3058 + 6f136a0
commit ebc0c25
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: