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', ],