Upper limit datasets version (#3824)

schustmi · web-flow · commit b21c22cbfbd0 · 2025-07-10T16:06:31.000+02:00
diff --git a/examples/quickstart/requirements.txt b/examples/quickstart/requirements.txt
@@ -1,7 +1,7 @@
 zenml[server]==0.83.1
 notebook
 pyarrow
-datasets
+datasets<4.0.0
 transformers
 transformers[torch]
 torch
diff --git a/examples/quickstart/requirements_aws.txt b/examples/quickstart/requirements_aws.txt
@@ -1,7 +1,7 @@
 zenml[server]==0.83.1
 notebook
 pyarrow
-datasets
+datasets<4.0.0
 transformers
 transformers[torch]
 torch
diff --git a/examples/quickstart/requirements_azure.txt b/examples/quickstart/requirements_azure.txt
@@ -1,7 +1,7 @@
 zenml[server]==0.83.1
 notebook
 pyarrow
-datasets
+datasets<4.0.0
 transformers
 transformers[torch]
 torch
diff --git a/examples/quickstart/requirements_gcp.txt b/examples/quickstart/requirements_gcp.txt
@@ -1,7 +1,7 @@
 zenml[server]==0.83.1
 notebook
 pyarrow
-datasets
+datasets<4.0.0
 transformers
 transformers[torch]
 torch
diff --git a/src/zenml/integrations/huggingface/__init__.py b/src/zenml/integrations/huggingface/__init__.py
@@ -49,7 +49,7 @@ def get_requirements(cls, target_os: Optional[str] = None, python_version: Optio
             A list of requirements.
         """
         requirements = [
-            "datasets>=2.16.0",
+            "datasets>=2.16.0,<4.0.0",
             "huggingface_hub>0.19.0",
             "accelerate",
             "bitsandbytes>=0.41.3",
