From ea84dde71a7e965954b2bc3dba1f866daa4bac38 Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Fri, 23 Aug 2024 14:21:53 +0200
Subject: [PATCH] better version ranges to reflect max and min transformers
 support

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

diff --git a/setup.py b/setup.py
index 68a25244ca..cd0306322d 100644
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,7 @@
 REQUIRED_PKGS = [
     "coloredlogs",
     "sympy",
-    "transformers[sentencepiece]>=4.29.0,<4.45.0",
+    "transformers[sentencepiece]>=4.29,<=4.44",
     "torch>=1.11",
     "packaging",
     "numpy<2.0",  # transformers requires numpy<2.0 https://github.com/huggingface/transformers/pull/31569
@@ -24,6 +24,7 @@
 ]
 
 # TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released
+# pytest>=8.0.0 also fails with the transformers version pinned for exporters-tf
 TESTS_REQUIRE = [
     "accelerate",
     "pytest<=8.0.0",
@@ -72,7 +73,7 @@
         "timm",
         "h5py",
         "numpy<1.24.0",
-        "transformers[sentencepiece]>=4.26.0,<4.38.0",
+        "transformers[sentencepiece]>=4.26,<4.38",
     ],
     "diffusers": ["diffusers"],
     "intel": "optimum-intel>=1.18.0",