Restrict tensorflow requirements

ZwickyTransientFacility · Feb 23, 2024 · db52661 · db52661
1 parent 2720bed
commit db52661
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ keras-tuner = ">=1.0.2"
 matplotlib = ">=3.3"
 questionary = ">=1.8.1"
 scikit-learn = ">=0.24.1"
-tensorflow = ">=2.14.0"
+tensorflow = ">=2.14.0,<=2.15.0"
 wandb = ">=0.12.1"
 h5py = ">=3.10.0"
 astropy = ">=5.2.2"

diff --git a/requirements.txt b/requirements.txt
@@ -23,5 +23,5 @@ keras-tuner>=1.0.2
 matplotlib>=3.3
 questionary>=1.8.1
 scikit-learn>=0.24.1
-tensorflow>=2.14.0
+tensorflow>=2.14.0,<=2.15.0
 wandb>=0.12.1