Add evaluate to test dependencies (#18396)

2022-08-01 08:55:44 -04:00
parent bd6d1b4300
commit af1e6b4d87
2 changed files with 3 additions and 0 deletions
--- 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",