diff --git a/setup.py b/setup.py index c861731a..f8347f33 100644 --- a/setup.py +++ b/setup.py @@ -54,6 +54,7 @@ "huggingface-hub>=0.7.0", # Utilities from PyPA to e.g., compare versions "packaging", + "scikit-learn", ] TEMPLATE_REQUIRE = [ @@ -89,7 +90,6 @@ "sacremoses", "scipy", "seqeval", - "scikit-learn", "jiwer", "sentencepiece", # for bleurt "transformers", # for evaluator