Moving metrics to extended tasks.

huggingface · Dec 11, 2024 · bab94af · bab94af
1 parent 5a8ca46
commit bab94af
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -74,8 +74,6 @@ dependencies = [
     "scikit-learn",
     "spacy==3.7.2",
     "sacrebleu",
-    "unbabel-comet",
-    "gemba",
     "rouge_score==0.1.2",
     "sentencepiece>=0.1.99",
     "protobuf==3.20.*", # pinned for sentencepiece compat
@@ -101,7 +99,9 @@ docs = ["hf-doc-builder", "watchdog"]
 extended_tasks = [
   "langdetect", # ifeval
   "openai", # llm as a judge using openai models
-  "tiktoken"
+  "tiktoken",
+  "unbabel-comet",
+  "gemba",
 ]
 s3 = ["s3fs"]
 multilingual = [