Refuse Datasets 2.5.0 while waiting for a patch

2022-09-21 09:37:53 -04:00
parent 486134e5a0
commit 114295c010
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -102,7 +102,7 @@ _deps = [
    "codecarbon==1.2.0",
    "cookiecutter==1.7.3",
    "dataclasses",
-    "datasets",
+    "datasets!=2.5.0",
    "deepspeed>=0.6.5",
    "dill<0.3.5",
    "evaluate>=0.2.0",
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -8,7 +8,7 @@ deps = {
    "codecarbon": "codecarbon==1.2.0",
    "cookiecutter": "cookiecutter==1.7.3",
    "dataclasses": "dataclasses",
-    "datasets": "datasets",
+    "datasets": "datasets!=2.5.0",
    "deepspeed": "deepspeed>=0.6.5",
    "dill": "dill<0.3.5",
    "evaluate": "evaluate>=0.2.0",