resolved issue with sklearn version >= 2.3.x

deephyper · Nov 6, 2020 · aaf97ff · aaf97ff
1 parent f76ba4d
commit aaf97ff
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/skopt/__init__.py b/skopt/__init__.py
@@ -29,7 +29,7 @@
 # Dev branch marker is: 'X.Y.dev' or 'X.Y.devN' where N is an integer.
 # 'X.Y.dev0' is the canonical version of 'X.Y.dev'
 #
-__version__ = "0.8.1"
+__version__ = "0.8.2"
 
 if __SKOPT_SETUP__:
     import sys

diff --git a/skopt/space/space.py b/skopt/space/space.py
@@ -11,6 +11,10 @@
 from sklearn.utils import check_random_state
 from sklearn.utils.fixes import sp_version
 
+if type(sp_version) is not tuple:  # Version object since sklearn>=2.3.x
+    sp_version = sp_version.release
+
+
 from .transformers import CategoricalEncoder
 from .transformers import StringEncoder
 from .transformers import LabelEncoder