Test fix

pytorch · Feb 6, 2025 · 3b1f97a · 3b1f97a
1 parent 1d75c8f
commit 3b1f97a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build_wheels_aarch64_linux.yml b/.github/workflows/build_wheels_aarch64_linux.yml
@@ -29,7 +29,7 @@ jobs:
       test-infra-repository: pytorch/test-infra
       test-infra-ref: main
       with-cuda: disable
-
+      python-versions: '[\"3.9\", \"3.10\"]'
   build:
     needs: generate-matrix
     permissions:

diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh
@@ -15,6 +15,6 @@ if [ -z ${PYTORCH_VERSION:-} ]; then
 else
     PYTORCH_DEP="torch==$PYTORCH_VERSION"
 fi
-pip install $PYTORCH_DEP
+pip3 install --pre torch  --index-url https://download.pytorch.org/whl/nightly/cpu
 
 pip install setuptools wheel twine auditwheel