Add evaluate to test dependencies (#18396)

af1e6b4d · Sylvain Gugger · GitHub · bd6d1b43 · af1e6b4d · af1e6b4d
Unverified Commit af1e6b4d authored Aug 01, 2022 by Sylvain Gugger Committed by GitHub Aug 01, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

setup.py setup.py +2 -0

src/transformers/dependency_versions_table.py src/transformers/dependency_versions_table.py +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -105,6 +105,7 @@ _deps = [
    "datasets",
    "deepspeed>=0.6.5",
    "dill<0.3.5",
+    "evaluate",
    "fairscale>0.3",
    "faiss-cpu",
    "fastapi",
@@ -289,6 +290,7 @@ extras["testing"] = (
        "psutil",
        "datasets",
        "dill",
+        "evaluate",
        "pytest-timeout",
        "black",
        "sacrebleu",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -11,6 +11,7 @@ deps = {
    "datasets": "datasets",
    "deepspeed": "deepspeed>=0.6.5",
    "dill": "dill<0.3.5",
+    "evaluate": "evaluate",
    "fairscale": "fairscale>0.3",
    "faiss-cpu": "faiss-cpu",
    "fastapi": "fastapi",