Add version restrictions to tensorflow for pypi

talmolab · Sep 11, 2023 · a4d2da8 · a4d2da8
1 parent e4fca4f
commit a4d2da8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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'