updated tensorflow requirement.

PatReis · PatReis · commit eb034a39e024 · 2024-05-06T11:36:59.000+02:00
diff --git a/docs/environment.yml b/docs/environment.yml
@@ -13,7 +13,7 @@ dependencies:
   - pandoc
   - pip:
     - keras>=3.0.0
-    - tf-nightly-cpu==2.16.0.dev20231103  # Only need tensorflow for docs
+    - tensorflow-cpu~=2.16.1  # Only need tensorflow for docs
     - jinja2==3.0.3
     - numpy
     - scipy
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,5 +1,5 @@
 keras>=3.0.0
-tf-nightly-cpu==2.16.0.dev20231103
+tensorflow-cpu~=2.16.1
 sphinx==3.4.3
 sphinx_rtd_theme
 jinja2==3.0.0
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 keras>=3.0.0
 torch>=2.1.0
-tf-nightly-cpu==2.16.0.dev20240101
+tensorflow-cpu~=2.16.1
 torchvision>=0.16.0
 jax[cpu]
 numpy>=1.23.0
diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
         # "dm-tree",
         "keras>=3.0.2",
         # Backends
-        # "tf-nightly-cpu==2.16.0.dev20240101",
+        # "tensorflow-cpu>=2.16.1",
         # "torch>=2.1.0",
         # "torchvision>=0.16.0",
         # "jax[cpu]",
