From 33abeb1225d6697abbcc3af930859dabab82c2aa Mon Sep 17 00:00:00 2001
From: Madhur Karampudi <vkarampudi@google.com>
Date: Fri, 15 Mar 2024 10:57:46 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 616183026
---
 setup.py                    | 2 +-
 tensorboard_plugin/setup.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 90ef01e..76e0832 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,7 @@ def select_constraint(default, nightly=None, git_master=None):
     return default
 
 REQUIRED_PACKAGES = [
-    'tensorflow>=2.15,<3',
+    'tensorflow~=2.15',
     'tensorflow-hub>=0.16.1,<1.0.0',
     'tensorflow-data-validation' + select_constraint(
         default='>=1.14.0,<1.15.0',
diff --git a/tensorboard_plugin/setup.py b/tensorboard_plugin/setup.py
index 2c1c50b..d6a9542 100644
--- a/tensorboard_plugin/setup.py
+++ b/tensorboard_plugin/setup.py
@@ -44,7 +44,7 @@ def select_constraint(default, nightly=None, git_master=None):
 REQUIRED_PACKAGES = [
     'protobuf>=3.20.3,<5',
     'tensorboard>=2.15.2,<2.16.0',
-    'tensorflow>=2.15,<3',
+    'tensorflow~=2.15',
     'tensorflow-model-analysis'
     + select_constraint(
         default='>=0.45,<=0.46',