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 type annotations in python_tests

Browse Source
This commit is contained in:
c-bata
2024-07-23 16:49:37 +09:00
parent 0198825ded
commit 640f83ecff
8 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/streamlit/test_streamlit_helper.py
+4 -3
View File
@@ -1,6 +1,7 @@
from __future__ import annotations
import itertools
from typing import Sequence
from typing import Union
import optuna
from optuna_dashboard import ChoiceWidget
@@ -60,7 +61,7 @@ for r in range(len(widget_list) + 1):
@pytest.mark.parametrize("widgets", widgets_combinations_for_user_attr)
def test_render_user_attr_form_widgets(
widgets: Sequence[Union[ChoiceWidget, SliderWidget, TextInputWidget]],
widgets: Sequence[ChoiceWidget | SliderWidget | TextInputWidget],
) -> None:
study = optuna.create_study()
register_user_attr_form_widgets(study, widgets) # type: ignore
@@ -77,7 +78,7 @@ for r in range(1, len(widget_list) + 1):
@pytest.mark.parametrize("widgets", widgets_combinations_for_objective)
def test_render_objective_form_widgets(
widgets: Sequence[Union[ChoiceWidget, SliderWidget, TextInputWidget]],
widgets: Sequence[ChoiceWidget | SliderWidget | TextInputWidget],
) -> None:
study = optuna.create_study(directions=["maximize"] * len(widgets))
register_objective_form_widgets(study, widgets) # type: ignore