diff --git a/setup.py b/setup.py index f5db4639..eb0d94de 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,6 @@ import setuptools -tensorflow = ["tensorflow>=2.6"] +tensorflow = ["tensorflow>=2.6,<2.16.0"] pytorch = ["torch>=1.11", "torchdata"] sklearn = ["scikit-learn>=1.0"] cloud = ["tiledb-cloud"]