Fix unit test with optuna master

keisuke-umezawa · Feb 6, 2024 · 522b6ff · 522b6ff
1 parent cfadf70
commit 522b6ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python_tests/test_cached_extra_study_property.py b/python_tests/test_cached_extra_study_property.py
@@ -132,7 +132,7 @@ def test_contains_failed_trials(self) -> None:
             create_trial(
                 state=TrialState.COMPLETE, value=0, distributions=distributions, params=params
             ),
-            create_trial(state=TrialState.FAIL, value=0, distributions={}, params={}),
+            create_trial(state=TrialState.FAIL, value=None, distributions={}, params={}),
             create_trial(
                 state=TrialState.COMPLETE, value=0, distributions=distributions, params=params
             ),
@@ -235,7 +235,7 @@ def test_contains_failed_trials(self) -> None:
             ),
             create_trial(
                 state=TrialState.FAIL,
-                value=0,
+                value=None,
                 distributions={},
                 params={},
                 user_attrs={"bar": "bar"},