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 18:15:30 +09:00
committed by GitHub
co-authored by Shuhei Watanabe
parent d7ff6f41f4
commit a8c451ada2
+3 -1
View File
@@ -107,4 +107,6 @@ def test_download_csv_user_attr() -> None:
study = optuna.create_study(storage=storage)
optuna.logging.set_verbosity(optuna.logging.ERROR)
study.optimize(objective, n_trials=10)
_validate_output(storage, 200, 0)
cols = ["Param x", "Param y", "Value", "UserAttribute abs_y"]
_validate_output(storage, 200, 0, extra_col_names=cols)