From a9b485224ff4ae78eda7458e9191f7434ef8edc5 Mon Sep 17 00:00:00 2001
From: Wei Wang <weiwan@nvidia.com>
Date: Wed, 11 Sep 2024 11:50:20 -0700
Subject: [PATCH] Limit the Workaround Scope

---
 .github/scripts/validate_binaries.sh | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
index 74012506a..5a0d950dd 100755
--- a/.github/scripts/validate_binaries.sh
+++ b/.github/scripts/validate_binaries.sh
@@ -15,9 +15,14 @@ else
         conda install -y conda=23.11.0
     fi
     # Please note ffmpeg is required for torchaudio, see https://github.com/pytorch/pytorch/issues/96159
-    conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} ffmpeg
+    conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
     conda activate ${ENV_NAME}
-    pip3 install numpy --force-reinstall
+    
+    # Remove when https://github.com/pytorch/builder/issues/1985 is fixed
+    if [[ ${MATRIX_GPU_ARCH_TYPE} == 'cuda' && ${TARGET_OS} == 'linux-aarch64' ]]; then 
+        pip3 install numpy --force-reinstall
+    fi
+
     INSTALLATION=${MATRIX_INSTALLATION/"conda install"/"conda install -y"}
     TEST_SUFFIX=""