From dd54fda959e846b663ded40375b07215cae7b3a0 Mon Sep 17 00:00:00 2001 From: "Wang, Yi" Date: Thu, 12 Dec 2024 21:34:58 +0800 Subject: [PATCH] Fix scikit-learn to 1.5.2 to fix f1 evaluation crash in 1.6.0 (#1596) Signed-off-by: Wang, Yi A --- examples/image-classification/requirements.txt | 4 ++-- examples/language-modeling/requirements.txt | 2 +- examples/protein-folding/requirements.txt | 2 +- examples/text-classification/requirements.txt | 2 +- examples/trl/requirements.txt | 2 +- 5 files changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/image-classification/requirements.txt b/examples/image-classification/requirements.txt index 7b0e43a8d2..4cbf42532d 100644 --- 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 \ No newline at end of file +scikit-learn == 1.5.2 +timm>=0.9.16 diff --git a/examples/language-modeling/requirements.txt b/examples/language-modeling/requirements.txt index 4c2256d81b..aa223dd7f6 100644 --- 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 index 9caf8e887c..e3b2a3ba45 100644 --- 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 index d6536070ac..7ce7d0ba42 100644 --- 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 index 248692fd27..3a9be36241 100644 --- 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