modify with tensorflow-metal

Author: Steboss

.github/container/pip-finalize.sh

@@ -53,8 +53,8 @@ else
   # Replace any tensorflow==X with tensorflow-cpu==X in requirements.txt only on amd64
   if [[ "$(uname -m)" = "x86_64" ]]; then
     sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-cpu==\1/' requirements.txt
-  elif [[ "$(uname -m)" = "arm64" ]]; then
-    sed -i 's/^tensorflow==\([0-9.*]\+\)$/tensorflow-metal==1.1.0/' requirements.txt
+  elif [[ "$(uname -m)" = "arm64" ]] || [[ "$(uname -m)" = "aarch64" ]]; then
+    sed -i 's/^tensorflow==.*$/tensorflow-metal==1.1.0/' requirements.txt
   else
     echo "Skipping TF on $(uname -m)"
   fi