From 462bc99b339b49733852e7c8cd8e73f4df4ccb5f Mon Sep 17 00:00:00 2001 From: Hiroki Takizawa Date: Thu, 7 Dec 2023 10:31:31 +0900 Subject: [PATCH] Update python_tests/test_csv_download.py Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com> --- python_tests/test_csv_download.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/python_tests/test_csv_download.py b/python_tests/test_csv_download.py index 89f8978f..55cd4569 100644 --- a/python_tests/test_csv_download.py +++ b/python_tests/test_csv_download.py @@ -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)