From f05cbf0f6f2e961b2ba5f31d03db995930a9b548 Mon Sep 17 00:00:00 2001
From: tenzen-y <yuki.iwai.tz@gmail.com>
Date: Sun, 28 Aug 2022 18:57:36 +0900
Subject: [PATCH] fix build script

---
 scripts/v1beta1/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/v1beta1/build.sh b/scripts/v1beta1/build.sh
index 9354951a77f..0acc611e04a 100755
--- a/scripts/v1beta1/build.sh
+++ b/scripts/v1beta1/build.sh
@@ -140,7 +140,7 @@ else
 fi
 
 echo -e "\nBuilding mxnet mnist training container example...\n"
-docker buildx build --platform linux/amd64 -t "${REGISTRY}/mxnet-mnist:${TAG}" -f examples/${VERSION}/trial-images/mxnet-mnist/Dockerfile .
+docker buildx build --platform "linux/$ARCH" -t "${REGISTRY}/mxnet-mnist:${TAG}" -f examples/${VERSION}/trial-images/mxnet-mnist/Dockerfile .
 
 echo -e "\nBuilding Tensorflow with summaries mnist training container example...\n"
 docker buildx build --platform "linux/$ARCH" -t "${REGISTRY}/tf-mnist-with-summaries:${TAG}" -f examples/${VERSION}/trial-images/tf-mnist-with-summaries/Dockerfile .