From d7ff6f41f45fb1d8dbc5e61ff3642d7c99f52f5f Mon Sep 17 00:00:00 2001 From: Hiroki Takizawa Date: Thu, 7 Dec 2023 18:15:19 +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 | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python_tests/test_csv_download.py b/python_tests/test_csv_download.py index a93789e7..3eb0f330 100644 --- a/python_tests/test_csv_download.py +++ b/python_tests/test_csv_download.py @@ -90,7 +90,10 @@ def test_download_csv_multi_obj(is_multi_obj: bool) -> None: study = optuna.create_study(storage=storage, directions=directions) optuna.logging.set_verbosity(optuna.logging.ERROR) study.optimize(objective, n_trials=10) - _validate_output(storage, 200, 0) + cols = ["Param x", "Param y"] + cols += ["Objective 0", "Objective 1"] if is_multi_obj else ["Value"] + _validate_output(storage, 200, 0, extra_col_names=cols) + def test_download_csv_user_attr() -> None: