From 294c0657df2f2dd9a22b55363c7f532c872cbd2a Mon Sep 17 00:00:00 2001 From: cross32768 Date: Fri, 7 Jul 2023 16:28:05 +0900 Subject: [PATCH] Add default value for on_success_callback --- optuna_dashboard/streamlit/_streamlit_helper.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/optuna_dashboard/streamlit/_streamlit_helper.py b/optuna_dashboard/streamlit/_streamlit_helper.py index 14f746fd..252c0901 100644 --- a/optuna_dashboard/streamlit/_streamlit_helper.py +++ b/optuna_dashboard/streamlit/_streamlit_helper.py @@ -83,7 +83,9 @@ def _render_widgets( def render_user_attr_form_widgets( - study: optuna.Study, trial: FrozenTrial, on_success_callback: Optional[Callable[[], None]] + study: optuna.Study, + trial: FrozenTrial, + on_success_callback: Optional[Callable[[], None]] = None, ) -> None: """Render user input widgets to UI with streamlit. @@ -123,7 +125,9 @@ def render_user_attr_form_widgets( def render_objective_form_widgets( - study: optuna.Study, trial: FrozenTrial, on_success_callback: Optional[Callable[[], None]] + study: optuna.Study, + trial: FrozenTrial, + on_success_callback: Optional[Callable[[], None]] = None, ) -> None: """Render user input widgets to UI with streamlit.