From 8507560aa618859f91772acad625e0e5fd89b461 Mon Sep 17 00:00:00 2001
From: Nikita Savelyev <nikita.savelyev@intel.com>
Date: Mon, 17 Mar 2025 16:40:45 +0100
Subject: [PATCH] Add upper bound for datasets package

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 60851f39a..8c89237bf 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,8 @@
     "torch>=1.11",
     "optimum~=1.24",
     "transformers>=4.36,<4.50",
-    "datasets>=1.4.0",
+    # TODO: Remove the upper bound after the next datasets release (https://github.com/huggingface/datasets/issues/7458)
+    "datasets>=1.4.0,<3.4.0",
     "sentencepiece",
     "setuptools",
     "scipy",