Remove tensorflow-macos from setup.py

PiperOrigin-RevId: 615232009
tensorflow · Mar 13, 2024 · f0f675c · f0f675c
1 parent 9aaf8ac
commit f0f675c
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/oss_scripts/pip_package/setup.py b/oss_scripts/pip_package/setup.py
@@ -74,17 +74,12 @@ def finalize_options(self):
     distclass=BinaryDistribution,
     install_requires=[
         (
-            'tensorflow>=2.15.0, <2.16; platform_machine != "arm64" or'
-            ' platform_system != "Darwin"'
-        ),
-        (
-            'tensorflow-macos>=2.15.0, <2.16; platform_machine == "arm64" and'
-            ' platform_system == "Darwin"'
+            'tensorflow>=2.16.1, <2.17',
         ),
     ],
     extras_require={
         'tensorflow_cpu': [
-            'tensorflow-cpu>=2.15.0, <2.16',
+            'tensorflow-cpu>=2.16.1, <2.17',
         ],
         'tests': [
             'absl-py',