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

Format with black

Browse Source
This commit is contained in:
c-bata
2020-10-24 17:50:01 +09:00
parent 92b5bc4f7d
commit c46c630f7b
2 changed files with 43 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
optuna_dashboard/app.py
+37 -38
View File
@@ -25,10 +25,11 @@ def catch_api_internal_error(view):
return response_body
except Exception as e:
response.status = 500
response.content_type = 'application/json'
stacktrace = '\n'.join(traceback.format_tb(e.__traceback__))
response.content_type = "application/json"
stacktrace = "\n".join(traceback.format_tb(e.__traceback__))
_logger.error(f"Exception: {e}\n{stacktrace}")
return json.dumps({"reason": "internal server error"})
return decorated
@@ -52,29 +53,29 @@ def create_app(storage):
def dashboard():
response.content_type = "text/html"
return """<!DOCTYPE html>
<html lang="en">
<head>
<title>Optuna Dashboard</title>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1" />
<style>
body {
min-height: 100vh;
margin: 0;
padding: 0;
}
</style>
<script defer src="/static/bundle.js"></script>
</head>
<body>
<noscript>You need to enable JavaScript to run this dashboard.</noscript>
<div id="dashboard">
<p>Now loading...</p>
</div>
</body>
</html>"""
<html lang="en">
<head>
<title>Optuna Dashboard</title>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1" />
<style>
body {
min-height: 100vh;
margin: 0;
padding: 0;
}
</style>
<script defer src="/static/bundle.js"></script>
</head>
<body>
<noscript>You need to enable JavaScript to run this dashboard.</noscript>
<div id="dashboard">
<p>Now loading...</p>
</div>
</body>
</html>"""
@app.get("/api/studies")
@catch_api_internal_error
@@ -84,22 +85,22 @@ def create_app(storage):
serializer.serialize_study_summary(summary)
for summary in storage.get_all_study_summaries()
]
return json.dumps({
"study_summaries": summaries,
})
return json.dumps(
{
"study_summaries": summaries,
}
)
@app.post("/api/studies")
@catch_api_internal_error
def create_study():
response.content_type = 'application/json'
response.content_type = "application/json"
study_name = request.json.get("study_name", None)
direction = request.json.get("direction", None)
if study_name is None or direction not in ("minimize", "maximize"):
response.status = 400
return {
"reason": "You need to set study_name and direction"
}
return {"reason": "You need to set study_name and direction"}
study_id = storage.create_new_study(study_name)
if direction.lower() == "maximize":
@@ -107,9 +108,9 @@ def create_app(storage):
summary = get_study_summary(storage, study_id)
response.status = 201
return json.dumps({
"study_summary": serializer.serialize_study_summary(summary)
})
return json.dumps(
{"study_summary": serializer.serialize_study_summary(summary)}
)
@app.get("/api/studies/<study_id:int>")
@catch_api_internal_error
@@ -117,9 +118,7 @@ def create_app(storage):
response.content_type = "application/json"
summary = get_study_summary(storage, study_id)
trials = storage.get_all_trials(study_id)
return json.dumps(serializer.serialize_study_detail(
summary, trials
))
return json.dumps(serializer.serialize_study_detail(summary, trials))
@app.get("/static/<filename:path>")
def send_static(filename):
optuna_dashboard/serializer.py
+6 -8
View File
@@ -27,7 +27,7 @@ def serialize_study_summary(summary: StudySummary) -> Dict[str, Any]:
}
if summary.datetime_start is not None:
serialized["datetime_start"] = summary.datetime_start.isoformat(),
serialized["datetime_start"] = (summary.datetime_start.isoformat(),)
if summary.best_trial: # Make undefined if None
serialized["best_trial"] = serialize_frozen_trial(summary.best_trial)
@@ -35,7 +35,9 @@ def serialize_study_summary(summary: StudySummary) -> Dict[str, Any]:
return serialized
def serialize_study_detail(summary: StudySummary, trials: List[FrozenTrial]) -> Dict[str, Any]:
def serialize_study_detail(
summary: StudySummary, trials: List[FrozenTrial]
) -> Dict[str, Any]:
serialized = {
"name": summary.study_name,
"direction": summary.direction.name.lower(),
@@ -44,13 +46,9 @@ def serialize_study_detail(summary: StudySummary, trials: List[FrozenTrial]) ->
serialized["datetime_start"] = summary.datetime_start.isoformat()
if summary.best_trial is not None:
serialized["best_trial"] = serialize_frozen_trial(
summary.best_trial)
serialized["best_trial"] = serialize_frozen_trial(summary.best_trial)
serialized["trials"] = [
serialize_frozen_trial(trial)
for trial in trials
]
serialized["trials"] = [serialize_frozen_trial(trial) for trial in trials]
return serialized