Temporarily pin datasets upper version to fix CI (huggingface#31407)

Temporarily pin datasets upper version
ylacombe · Jun 13, 2024 · b767282 · b767282
1 parent 67a4ef8
commit b767282
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -102,7 +102,7 @@
     "codecarbon==1.2.0",
     "cookiecutter==1.7.3",
     "dataclasses",
-    "datasets!=2.5.0",
+    "datasets!=2.5.0,<2.20.0",  # Temporary upper version
     "decord==0.6.0",
     "deepspeed>=0.9.3",
     "diffusers",

diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
@@ -9,7 +9,7 @@
     "codecarbon": "codecarbon==1.2.0",
     "cookiecutter": "cookiecutter==1.7.3",
     "dataclasses": "dataclasses",
-    "datasets": "datasets!=2.5.0",
+    "datasets": "datasets!=2.5.0,<2.20.0",
     "decord": "decord==0.6.0",
     "deepspeed": "deepspeed>=0.9.3",
     "diffusers": "diffusers",