Update python_tests/test_csv_download.py

Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com>
This commit is contained in:
Hiroki Takizawa
2023-12-07 10:31:31 +09:00
committed by GitHub
co-authored by Shuhei Watanabe
parent 64dc46c0c2
commit 462bc99b33
+2 -4
View File
@@ -86,10 +86,8 @@ def test_download_csv_multi_obj(is_multi_obj: bool) -> None:
return x**2 + y
storage = optuna.storages.InMemoryStorage()
if is_multi_obj:
study = optuna.create_study(storage=storage, directions=["minimize", "minimize"])
else:
study = optuna.create_study(storage=storage)
directions = ["minimize", "minimize"] if is_multi_obj else ["minimize"]
study = optuna.create_study(storage=storage, directions=directions)
optuna.logging.set_verbosity(optuna.logging.ERROR)
study.optimize(objective, n_trials=10)
app = create_app(storage)