Depends on 'tensorflow>=2.15,<2.16'.

PiperOrigin-RevId: 610814609
tensorflow · Feb 27, 2024 · d697414 · d697414
1 parent 6748bff
commit d697414
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/RELEASE.md b/RELEASE.md
@@ -12,7 +12,7 @@
     (previously was 16.04).
 *   Explicitly use Keras 2 or `tf_keras`` if Keras 3 is installed.
 *   Added python 3.11 support.
-*   Depends on `tensorflow>=2.15.0,<3`.
+*   Depends on `tensorflow>=2.15,<2.16`.
 *   Enable passing `tf.saved_model.SaveOptions` to model saving functionality.
 *   Census and sentiment examples updated to only use Keras instead of
     estimator.

diff --git a/setup.py b/setup.py
@@ -48,7 +48,9 @@ def _make_required_install_packages():
       'protobuf>=3.20.3,<5',
       'pyarrow>=10,<11',
       'pydot>=1.2,<2',
-      'tensorflow>=2.15,<3',
+      # TODO(b/303870331) Update to '>=2.16,<3' once the rest of the ecosystem
+      # updates and the incompatibilities are fixed.
+      'tensorflow>=2.15,<2.16',
       'tensorflow-metadata' + select_constraint(
           default='>=1.14.0,<1.15.0',
           nightly='>=1.15.0.dev',