From 0582cfdd1ab773bd53cf1f27dbf12333690d829f Mon Sep 17 00:00:00 2001 From: Steve Chien Date: Mon, 15 Apr 2024 15:33:30 -0700 Subject: [PATCH] Set TensorFlow version to at most 2.15.0 to avoid problems with using Keras 3. PiperOrigin-RevId: 625104162 --- requirements.txt | 2 +- setup.py | 2 +- setup_empirical.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index eb5c6bb..11871ec 100644 --- a/requirements.txt +++ b/requirements.txt @@ -39,5 +39,5 @@ statsmodels==0.14.0 tensorflow-datasets~=4.5 tensorflow-estimator~=2.4 tensorflow-probability~=0.22.0 -tensorflow~=2.4 +tensorflow>=2.4.0,<=2.15.0 tf-models-official~=2.13 diff --git a/setup.py b/setup.py index 34a958f..41d6c7c 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ setuptools.setup( 'packaging~=22.0', 'scikit-learn>=1.0,==1.*', 'scipy~=1.9', - 'tensorflow-estimator~=2.4', + 'tensorflow>=2.4.0,<=2.15.0', 'tensorflow-probability~=0.22.0', 'tensorflow~=2.4', ], diff --git a/setup_empirical.py b/setup_empirical.py index 7eed770..76c579f 100644 --- a/setup_empirical.py +++ b/setup_empirical.py @@ -43,7 +43,7 @@ setuptools.setup( 'scikit-learn>=1.0,==1.*', 'scipy~=1.9', 'statsmodels==0.14.0', - 'tensorflow~=2.4', + 'tensorflow>=2.4.0,<=2.15.0', 'tensorflow-privacy>=0.9.0', 'tf-models-official~=2.13', ],