From d697414b6f45d0a1f8364af7f9f7c42bde6929ba Mon Sep 17 00:00:00 2001
From: iindyk <iindyk@google.com>
Date: Tue, 27 Feb 2024 11:15:52 -0800
Subject: [PATCH] Depends on 'tensorflow>=2.15,<2.16'.

PiperOrigin-RevId: 610814609
---
 RELEASE.md | 2 +-
 setup.py   | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/RELEASE.md b/RELEASE.md
index 791cf59..9af9f0d 100644
--- 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
index 5384023..ce46be9 100644
--- 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',