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

Set include_best_trial=True in get_study_summary

Browse Source
This commit is contained in:
Ellon M
2022-03-28 20:26:26 +03:00
parent ba96fd624b
commit 6c8fc37c67
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
optuna_dashboard/_app.py
+5 -2
View File
@@ -77,7 +77,10 @@ def json_api_view(view: BottleView) -> BottleView:
def get_study_summary(storage: BaseStorage, study_id: int) -> Optional[StudySummary]:
summaries = storage.get_all_study_summaries()
if version.parse(optuna_ver) >= version.Version("3.0.0b0.dev"):
summaries = storage.get_all_study_summaries(include_best_trial=True)
else:
summaries = storage.get_all_study_summaries()
for summary in summaries:
if summary._study_id != study_id:
continue
@@ -123,7 +126,7 @@ def create_app(storage: BaseStorage, debug: bool = False) -> Bottle:
@app.get("/api/studies")
@json_api_view
def list_study_summaries() -> BottleViewReturn:
if version.parse(optuna_ver) >= version.Version("3.0.0b0"):
if version.parse(optuna_ver) >= version.Version("3.0.0b0.dev"):
summaries = storage.get_all_study_summaries(include_best_trial=True)
else:
summaries = storage.get_all_study_summaries()