Fix scikit-learn to 1.5.2 to fix f1 evaluation crash in 1.6.0 (huggin…

…gface#1596) Signed-off-by: Wang, Yi A <[email protected]>
skavulya · Dec 12, 2024 · dd54fda · dd54fda
1 parent f031d2d
commit dd54fda
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/examples/image-classification/requirements.txt b/examples/image-classification/requirements.txt
@@ -2,5 +2,5 @@ torch>=1.5.0
 torchvision>=0.6.0
 datasets>=2.14.0
 evaluate
-scikit-learn
-timm>=0.9.16
+scikit-learn == 1.5.2
+timm>=0.9.16
diff --git a/examples/language-modeling/requirements.txt b/examples/language-modeling/requirements.txt
@@ -2,5 +2,5 @@ datasets >= 2.14.0
 sentencepiece != 0.1.92
 protobuf
 evaluate
-scikit-learn
+scikit-learn == 1.5.2
 peft == 0.12.0
diff --git a/examples/protein-folding/requirements.txt b/examples/protein-folding/requirements.txt
@@ -1,2 +1,2 @@
 datasets>=2.14.0
-scikit-learn
+scikit-learn == 1.5.2
diff --git a/examples/text-classification/requirements.txt b/examples/text-classification/requirements.txt
@@ -1,7 +1,7 @@
 datasets >= 2.4.0
 sentencepiece != 0.1.92
 scipy
-scikit-learn
+scikit-learn == 1.5.2
 protobuf
 torch >= 1.3
 evaluate
diff --git a/examples/trl/requirements.txt b/examples/trl/requirements.txt
@@ -3,4 +3,4 @@ peft == 0.12.0
 datasets == 2.19.2
 tyro
 evaluate
-scikit-learn
+scikit-learn == 1.5.2