diff --git a/python_tests/test_api.py b/python_tests/test_api.py index 6828b83d..51017ac8 100644 --- a/python_tests/test_api.py +++ b/python_tests/test_api.py @@ -8,8 +8,8 @@ from optuna import get_all_study_summaries from optuna.study import StudyDirection from optuna_dashboard._app import create_app from optuna_dashboard._app import create_new_study -from optuna_dashboard.preferential import create_study from optuna_dashboard._preference_setting import register_output_component +from optuna_dashboard.preferential import create_study from .wsgi_client import send_request diff --git a/python_tests/test_preference_setting.py b/python_tests/test_preference_setting.py index 05727a04..033a9092 100644 --- a/python_tests/test_preference_setting.py +++ b/python_tests/test_preference_setting.py @@ -1,13 +1,11 @@ from __future__ import annotations + from unittest import TestCase import optuna - -from optuna_dashboard._preference_setting import ( - register_output_component, - _SYSTEM_ATTR_FEEDBACK_ARTIFACT_KEY, - _SYSTEM_ATTR_FEEDBACK_COMPONENT_TYPE, -) +from optuna_dashboard._preference_setting import _SYSTEM_ATTR_FEEDBACK_ARTIFACT_KEY +from optuna_dashboard._preference_setting import _SYSTEM_ATTR_FEEDBACK_COMPONENT_TYPE +from optuna_dashboard._preference_setting import register_output_component from optuna_dashboard.preferential._study import PreferentialStudy