Specify test-infra branch to main (#8046)

.github/workflows/build-cmake.yml

@@ -26,6 +26,7 @@ jobs:
       runner: ${{ matrix.runner }}
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -46,6 +47,7 @@ jobs:
     with:
       repository: pytorch/vision
       runner: ${{ matrix.runner }}
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -71,6 +73,7 @@ jobs:
       runner: ${{ matrix.runner }}
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
+      test-infra-ref: main
       script: |
         set -euo pipefail
 

.github/workflows/docs.yml

@@ -18,6 +18,7 @@ jobs:
     with:
       repository: pytorch/vision
       upload-artifact: docs
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -85,6 +86,7 @@ jobs:
       repository: pytorch/vision
       download-artifact: docs
       ref: gh-pages
+      test-infra-ref: main
       script: |
         set -euo pipefail
 

.github/workflows/lint.yml

@@ -14,6 +14,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       repository: pytorch/vision
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -40,6 +41,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       repository: pytorch/vision
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -74,6 +76,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       repository: pytorch/vision
+      test-infra-ref: main
       script: |
         set -euo pipefail
 

.github/workflows/tests.yml

@@ -33,6 +33,7 @@ jobs:
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
       timeout: 120
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -62,6 +63,7 @@ jobs:
       # and needs roughly 2 hours to just run the test suite
       timeout: 240
       runner: ${{ matrix.runner }}
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -94,6 +96,7 @@ jobs:
       gpu-arch-type: ${{ matrix.gpu-arch-type }}
       gpu-arch-version: ${{ matrix.gpu-arch-version }}
       timeout: 120
+      test-infra-ref: main
       script: |
         set -euxo pipefail
 
@@ -109,6 +112,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       repository: pytorch/vision
+      test-infra-ref: main
       script: |
         set -euo pipefail
 
@@ -139,6 +143,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       repository: pytorch/vision
+      test-infra-ref: main
       script: |
         set -euo pipefail
 

.github/workflows/update-viablestrict.yml

@@ -18,6 +18,7 @@ jobs:
     with:
       repository: pytorch/vision
       required_checks: "Build Linux,Build M1,Build Macos,Build Windows,Tests,CMake,Lint,Docs"
+      test-infra-ref: main
     secrets:
       ROCKSET_API_KEY: ${{ secrets.ROCKSET_API_KEY }}
       GITHUB_DEPLOY_KEY : ${{ secrets.VISION_GITHUB_DEPLOY_KEY }}