From 7608f1bb9605ef3d52c6d5e09226d2ec110a21a8 Mon Sep 17 00:00:00 2001 From: Contramundum Date: Mon, 4 Sep 2023 18:21:38 +0900 Subject: [PATCH] Fix test --- python_tests/test_api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python_tests/test_api.py b/python_tests/test_api.py index ae50e29a..c0304a37 100644 --- a/python_tests/test_api.py +++ b/python_tests/test_api.py @@ -102,7 +102,7 @@ class APITestCase(TestCase): def test_get_best_trials_of_preferential_study(self) -> None: storage = optuna.storages.InMemoryStorage() - study = create_study(storage=storage) + study = create_study(n_generate=4, storage=storage) for _ in range(3): trial = study.ask() study.mark_comparison_ready(trial) @@ -125,7 +125,7 @@ class APITestCase(TestCase): def test_report_preference(self) -> None: storage = optuna.storages.InMemoryStorage() - study = create_study(storage=storage) + study = create_study(n_generate=4, storage=storage) for _ in range(3): trial = study.ask() study.mark_comparison_ready(trial) @@ -153,7 +153,7 @@ class APITestCase(TestCase): def test_skip_trial(self) -> None: storage = optuna.storages.InMemoryStorage() - study = create_study(storage=storage) + study = create_study(n_generate=4, storage=storage) trials: list[optuna.Trial] = [] for _ in range(3): trial = study.ask()