fixes issue #1109 wrong bound on sklearn for sparse argument (#1112)

aeon-toolkit · Jan 26, 2024 · 73fe1b0 · 73fe1b0
1 parent d4c98ed
commit 73fe1b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aeon/classification/deep_learning/base.py b/aeon/classification/deep_learning/base.py
@@ -145,7 +145,7 @@ def convert_y_to_keras(self, y):
         installed_version = sklearn.__version__
         # Compare the installed version with the target version
         # categories='auto' to get rid of FutureWarning
-        if version.parse(installed_version) < version.parse("1.1"):
+        if version.parse(installed_version) < version.parse("1.2"):
             self.onehot_encoder = OneHotEncoder(sparse=False)
         else:
             self.onehot_encoder = OneHotEncoder(sparse_output=False)