Update Pandas import to new format

holgern · May 28, 2024 · d44fd88 · d44fd88
1 parent 7180081
commit d44fd88
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/skopt/space/space.py b/skopt/space/space.py
@@ -5,7 +5,12 @@
 import numpy as np
 import yaml
 from scipy.stats.distributions import randint, rv_discrete, uniform
-from sklearn.utils import check_pandas_support, check_random_state
+from sklearn.utils import check_random_state
+try:
+    # Syntax from sklearn 1.5.0 onwards
+    from sklearn.utils._optional_dependencies import check_pandas_support
+except ImportError:
+    from sklearn.utils import check_pandas_support
 
 from .transformers import (
     CategoricalEncoder,