From 5ec4b7b878d5215c7b75171644a3c70855ad1a01 Mon Sep 17 00:00:00 2001
From: Robert <robert.samoilescu@gmail.com>
Date: Thu, 28 Nov 2024 11:47:33 +0000
Subject: [PATCH] Upperbound tf to 2.16 because of keras 3 and transformers
 compatibility

---
 setup.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 452b912db..f6729a69e 100644
--- a/setup.py
+++ b/setup.py
@@ -17,13 +17,15 @@ def readme():
         'numba>=0.50.0, !=0.54.0, <0.60.0',  # Avoid 0.54 due to: https://github.com/SeldonIO/alibi/issues/466
     ],
 
-    'tensorflow': ['tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.19.0'],
+    # `keras 3` becomes the default for `tensorflow >= 2.16.0``
+    # which is not yet supported by `transformers`
+    'tensorflow': ['tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.16.0'],
     'torch': ['torch>=1.9.0, <3.0.0'],
     'all': [
         'ray>=0.8.7, <3.0.0',
         'shap>=0.40.0, <0.44.0',
         'numba>=0.50.0, !=0.54.0, <0.60.0',
-        'tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.19.0',
+        'tensorflow>=2.0.0, !=2.6.0, !=2.6.1, <2.16.0',
         'torch>=1.9.0, <3.0.0'
     ]
 }