diff --git a/python_tests/test_search_space.py b/python_tests/test_search_space.py index 44b460ea..c03ed218 100644 --- a/python_tests/test_search_space.py +++ b/python_tests/test_search_space.py @@ -5,8 +5,8 @@ import warnings import optuna from optuna import create_trial -from optuna.distributions import UniformDistribution from optuna.distributions import BaseDistribution +from optuna.distributions import UniformDistribution from optuna.exceptions import ExperimentalWarning from optuna.trial import TrialState