fix install_protoc ARCH value when linux container is used

Signed-off-by: Matteo Mortari <[email protected]>
kubeflow · Apr 17, 2024 · 8bcd792 · 8bcd792
1 parent 53990f7
commit 8bcd792
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/install_protoc.sh b/scripts/install_protoc.sh
@@ -11,7 +11,9 @@ if [[ "$OSTYPE" == "darwin"* ]]; then
 fi
 ARCH=$(uname -m)
 if [[ "$ARCH" == "arm"* ]]; then
-  ARCH="aarch_64"
+  ARCH="aarch_64" # Mac
+elif [[ "$ARCH" == "aarch64" ]]; then
+  ARCH="aarch_64" # Linux/container
 elif [[ "$ARCH" == "s390x" ]]; then
   ARCH="s390_64"
 elif [[ "$ARCH" == "ppc64le" ]] ; then