diff --git a/setup.py b/setup.py index 050a1012b..3e12a15e6 100644 --- a/setup.py +++ b/setup.py @@ -17,13 +17,13 @@ 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.15.0'], + 'tensorflow': ['tensorflow!=2.6.0, !=2.6.1, >=2.16.2, <2.17.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.15.0', + 'tensorflow!=2.6.0, !=2.6.1, >=2.16.2, <2.17.0', 'torch>=1.9.0, <3.0.0' ] }