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

Temporarily pin datasets upper version

Temporarily pin datasets upper version to fix CI (#31407)
Temporarily pin datasets upper version
b7672826 · Albert Villanova del Moral · GitHub · 67a4ef89 · b7672826 · b7672826
Unverified Commit b7672826 authored Jun 13, 2024 by Albert Villanova del Moral Committed by GitHub Jun 13, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +1 -1

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

No files found.
--- a/setup.py
+++ b/setup.py
@@ -102,7 +102,7 @@ _deps = [
    "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",

--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -9,7 +9,7 @@ deps = {
    "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",