From 581fc822bd6c335f97bdd449819c21fd6c25950a Mon Sep 17 00:00:00 2001 From: ktsitsi Date: Fri, 23 Aug 2024 13:21:08 +0300 Subject: [PATCH] Pin the tensorflow version lower than 2.16 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"]