Add conditionally tensorflow-macos to setup.py (#77)

secondmind-labs · Jul 22, 2022 · 06f238b · 06f238b
1 parent 0c6877b
commit 06f238b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,10 @@
     "gpflow>=2.1",
     "numpy",
     "scipy",
-    "tensorflow>=2.5.0,<2.9.0",
+    "tensorflow>=2.5.0,<2.9.0; platform_system!='Darwin' or platform_machine!='arm64'",
+    # NOTE: Support of Apple Silicon MacOS platforms is in an experimental mode
+    "tensorflow-macos>=2.5.0,<2.9.0; platform_system=='Darwin' and platform_machine=='arm64'",
+    # NOTE: once we require tensorflow-probability>=0.12, we can remove our custom deepcopy handling
     "tensorflow-probability>=0.13.0,<0.17.0",
     "protobuf~=3.19.0"
 ]