From a4d2da856d7a613c07e202bd853c50190f0fb3da Mon Sep 17 00:00:00 2001 From: roomrys <38435167+roomrys@users.noreply.github.com> Date: Mon, 11 Sep 2023 13:39:40 -0700 Subject: [PATCH] Add version restrictions to tensorflow for pypi --- pypi_requirements.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pypi_requirements.txt b/pypi_requirements.txt index b18637c37..93b0b7555 100644 --- a/pypi_requirements.txt +++ b/pypi_requirements.txt @@ -31,5 +31,7 @@ scikit-image scikit-learn ==1.0.* scikit-video seaborn -tensorflow +tensorflow>=2.6.3,<2.9.0; platform_machine != 'arm64' tensorflow-hub +tensorflow-macos==2.9.2; sys_platform == 'darwin' and platform_machine == 'arm64' +tensorflow-metal==0.5.0; sys_platform == 'darwin' and platform_machine == 'arm64'