From 919eae64177985458eb2cfd6d06a6ac3f4bed57c Mon Sep 17 00:00:00 2001
From: Matthew Grossman <matthewryangrossman@gmail.com>
Date: Thu, 6 Jul 2023 16:11:19 -0700
Subject: [PATCH] Fix macos localdev by splitting tensorflow deps

---
 requirements.txt | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 2b2e8d1e..864e82b3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,9 @@ rdt>=1.2,<1.3
 sdv>=0.17,<0.18
 sentencepiece==0.1.97
 smart_open>=2.1.0,<6.0
-tensorflow==2.11.1
+tensorflow-macos==2.11.0; platform_machine=="arm64"
+tensorflow-metal==0.7.1; platform_machine=="arm64"
+tensorflow==2.11.1; platform_machine=="x86_64"
 tensorflow_estimator==2.11
 tensorflow_privacy==0.7.3
 tensorflow_probability==0.19.0