diff --git a/setup.py b/setup.py index 9f9a276b3..83db9c010 100644 --- a/setup.py +++ b/setup.py @@ -16,20 +16,17 @@ def readme(): 'shap>=0.40.0, <0.44.0', # versioning: https://github.com/SeldonIO/alibi/issues/333 'numba>=0.50.0, !=0.54.0, <0.60.0', # Avoid 0.54 due to: https://github.com/SeldonIO/alibi/issues/466 ], - - # `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.19.0', - 'tf-keras>=2.0.0,<2.19.0' + 'tensorflow>=2.16.0, <2.19.0', + 'tf-keras>=2.16.0, <2.19.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', - 'tf-keras>=2.0.0,<2.19.0', + 'tensorflow>=2.16.0, <2.19.0', + 'tf-keras>=2.16.0, <2.19.0', 'torch>=1.9.0, <3.0.0', ] }