Merge pull request #361 from ntumlgroup/fix-sklearn

Fixed sklearn version before GridSearchCV is fixed
ASUS-AICS · Feb 28, 2024 · 7106704 · 7106704
2 parents 88e59da + 067a852
commit 7106704
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,6 +2,6 @@ liblinear-multicore
 numba
 pandas>1.3.0
 PyYAML
-scikit-learn
+scikit-learn==1.2.2
 scipy
 tqdm
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = libmultilabel
-version = 0.6.0
+version = 0.6.1
 author = LibMultiLabel Team
 license = MIT License
 license_file = LICENSE
@@ -29,7 +29,7 @@ install_requires =
     numba
     pandas>1.3.0
     PyYAML
-    scikit-learn
+    scikit-learn==1.2.2
     scipy
     tqdm