From 37b7c8f8e47388c7cfafff529b5e963cc4e9a0c8 Mon Sep 17 00:00:00 2001 From: Raviteja Gorijala <36429068+rtg0795@users.noreply.github.com> Date: Mon, 11 Mar 2024 00:09:39 -0700 Subject: [PATCH] Fix tensorflow version for macos --- oss_scripts/pip_package/setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/oss_scripts/pip_package/setup.py b/oss_scripts/pip_package/setup.py index 90c61b6da..0780e7c34 100644 --- a/oss_scripts/pip_package/setup.py +++ b/oss_scripts/pip_package/setup.py @@ -74,11 +74,11 @@ def finalize_options(self): distclass=BinaryDistribution, install_requires=[ ( - 'tensorflow>=2.16.0, <2.17; platform_machine != "arm64" or' + 'tensorflow>=2.16.1, <2.17; platform_machine != "arm64" or' ' platform_system != "Darwin"' ), ( - 'tensorflow-macos>=2.16.0, <2.17; platform_machine == "arm64" and' + 'tensorflow-macos>=2.16.1, <2.17; platform_machine == "arm64" and' ' platform_system == "Darwin"' ), ],