fix for pytest

This commit is contained in:
moririn2528
2023-09-06 11:30:59 +09:00
parent b1ad5eeff3
commit a48f04d6ed
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -159,7 +159,7 @@ class APITestCase(TestCase):
def test_report_preference_when_typo_mode(self) -> None:
storage = optuna.storages.InMemoryStorage()
study = create_study(storage=storage)
study = create_study(storage=storage, n_generate=3)
for _ in range(3):
trial = study.ask()
study.mark_comparison_ready(trial)
+1 -1
View File
@@ -15,7 +15,7 @@ from .storage_supplier import StorageSupplier
@parametrize_storages
def test_report_and_get_choices(storage_supplier: Callable[[], StorageSupplier]) -> None:
with storage_supplier() as storage:
study = create_study(storage=storage)
study = create_study(storage=storage, n_generate=5)
for _ in range(5):
trial = study.ask()
trial.suggest_float("x", 0, 1)