From f52519a88915b9cb9bfd648ab9124759a7884958 Mon Sep 17 00:00:00 2001
From: Antonio Sanchez <cantonios@google.com>
Date: Mon, 27 Nov 2023 14:27:55 -0800
Subject: [PATCH] Update tf-text setup scripts.

Now needs TF 2.15, and supports Python 3.11.

PiperOrigin-RevId: 585761002
---
 oss_scripts/pip_package/setup.nightly.py | 3 ++-
 oss_scripts/pip_package/setup.py         | 9 +++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/oss_scripts/pip_package/setup.nightly.py b/oss_scripts/pip_package/setup.nightly.py
index 04f284c0f..f5b67c00a 100644
--- a/oss_scripts/pip_package/setup.nightly.py
+++ b/oss_scripts/pip_package/setup.nightly.py
@@ -73,7 +73,7 @@ def finalize_options(self):
     cmdclass={'install': InstallPlatlib},
     distclass=BinaryDistribution,
     install_requires=[
-        'tensorflow_hub>=0.8.0',
+        'tensorflow_hub>=0.15.0',
     ],
     extras_require={
         'tests': [
@@ -92,6 +92,7 @@ def finalize_options(self):
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
         'Programming Language :: Python :: 3 :: Only',
         'Topic :: Scientific/Engineering',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',
diff --git a/oss_scripts/pip_package/setup.py b/oss_scripts/pip_package/setup.py
index bbe9f81d9..286675f33 100644
--- a/oss_scripts/pip_package/setup.py
+++ b/oss_scripts/pip_package/setup.py
@@ -74,18 +74,18 @@ def finalize_options(self):
     distclass=BinaryDistribution,
     install_requires=[
         (
-            'tensorflow>=2.13.0, <2.14; platform_machine != "arm64" or'
+            'tensorflow>=2.15.0, <2.16; platform_machine != "arm64" or'
             ' platform_system != "Darwin"'
         ),
         (
-            'tensorflow-macos>=2.13.0, <2.14; platform_machine == "arm64" and'
+            'tensorflow-macos>=2.15.0, <2.16; platform_machine == "arm64" and'
             ' platform_system == "Darwin"'
         ),
-        'tensorflow_hub>=0.13.0',
+        'tensorflow_hub>=0.15.0',
     ],
     extras_require={
         'tensorflow_cpu': [
-            'tensorflow-cpu>=2.13.0, <2.14',
+            'tensorflow-cpu>=2.15.0, <2.16',
         ],
         'tests': [
             'absl-py',
@@ -102,6 +102,7 @@ def finalize_options(self):
         'Programming Language :: Python :: 3',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
         'Programming Language :: Python :: 3 :: Only',
         'Topic :: Scientific/Engineering',
         'Topic :: Scientific/Engineering :: Artificial Intelligence',