wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update python_tests/test_csv_download.py

Browse Source 
Co-authored-by: Shuhei Watanabe <47781922+nabenabe0928@users.noreply.github.com>
This commit is contained in:
Hiroki Takizawa
2023-12-07 18:15:19 +09:00
committed by GitHub
co-authored by Shuhei Watanabe
parent 1e5d582572
commit d7ff6f41f4
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_csv_download.py
+4 -1
View File
@@ -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: