From 0592bb63446f8129e3f98b2425faccde7db4a7db Mon Sep 17 00:00:00 2001
From: Huy Do <huydhn@gmail.com>
Date: Wed, 12 Feb 2025 09:56:10 -0800
Subject: [PATCH] Fix test failures

---
 .../test_build_wheels_linux_without_cuda.yml     |  5 ++++-
 .github/workflows/test_build_wheels_m1.yml       | 16 ----------------
 .../test_build_wheels_windows_with_cuda.yml      |  5 -----
 3 files changed, 4 insertions(+), 22 deletions(-)

diff --git a/.github/workflows/test_build_wheels_linux_without_cuda.yml b/.github/workflows/test_build_wheels_linux_without_cuda.yml
index 30a18e3b25..c28fc74fc7 100644
--- a/.github/workflows/test_build_wheels_linux_without_cuda.yml
+++ b/.github/workflows/test_build_wheels_linux_without_cuda.yml
@@ -44,4 +44,7 @@ jobs:
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       package-name: ${{ matrix.package-name }}
       trigger-event: "${{ github.event_name }}"
-      build-platform: 'python-build-package'
+      build-platform: python-build-package
+      pip-install-torch-extra-args:
+        torchvision
+        torchao
diff --git a/.github/workflows/test_build_wheels_m1.yml b/.github/workflows/test_build_wheels_m1.yml
index f78e88651a..9bc3cc8f83 100644
--- a/.github/workflows/test_build_wheels_m1.yml
+++ b/.github/workflows/test_build_wheels_m1.yml
@@ -22,7 +22,6 @@ jobs:
       os: macos-arm64
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
-
   test:
     needs: generate-matrix
     strategy:
@@ -56,18 +55,3 @@ jobs:
       trigger-event: "${{ github.event_name }}"
       cache-path: ${{ matrix.cache-path }}
       cache-key: ${{ matrix.cache-key }}
-
-  test-artifact-name:
-    needs: generate-matrix
-    uses: ./.github/workflows/build_wheels_macos.yml
-    name: pytorch/ao
-    with:
-      repository: pytorch/ao
-      ref: drisspg/stack/31
-      test-infra-repository: pytorch/test-infra
-      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
-      pre-script: packaging/pre_build_script.sh
-      smoke-test-script: test/smoke_test.py
-      runner-type: macos-m1-stable
-      package-name: torchao
-      trigger-event: "${{ github.event_name }}"
diff --git a/.github/workflows/test_build_wheels_windows_with_cuda.yml b/.github/workflows/test_build_wheels_windows_with_cuda.yml
index 9520c45cc6..828a9280e8 100644
--- a/.github/workflows/test_build_wheels_windows_with_cuda.yml
+++ b/.github/workflows/test_build_wheels_windows_with_cuda.yml
@@ -28,11 +28,6 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - repository: pytorch/audio
-          env-script: packaging/vc_env_helper.bat
-          wheel-build-params: "--plat-name win_amd64"
-          smoke-test-script: test/smoke_test/smoke_test.py
-          package-name: torchaudio
         - repository: pytorch/vision
           pre-script: packaging/pre_build_script.sh
           env-script: packaging/windows/internal/vc_env_helper.bat