From 291c9aa2f3eb8702d1a6b153137823d4b0b0184b Mon Sep 17 00:00:00 2001
From: Huy Do <huydhn@gmail.com>
Date: Fri, 13 Oct 2023 14:26:49 -0700
Subject: [PATCH] [Release only] Pin test-infra repo for 2.1 release (#8045)

---
 .github/workflows/build-cmake.yml         | 3 +++
 .github/workflows/docs.yml                | 2 ++
 .github/workflows/lint.yml                | 3 +++
 .github/workflows/tests.yml               | 5 +++++
 .github/workflows/update-viablestrict.yml | 1 +
 5 files changed, 14 insertions(+)

diff --git a/.github/workflows/build-cmake.yml b/.github/workflows/build-cmake.yml
index 23f2b4b06ec..97c417ad4eb 100644
--- a/.github/workflows/build-cmake.yml
+++ b/.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: release/2.1
       script: |
         set -euo pipefail
 
@@ -46,6 +47,7 @@ jobs:
     with:
       repository: pytorch/vision
       runner: ${{ matrix.runner }}
+      test-infra-ref: release/2.1
       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: release/2.1
       script: |
         set -euo pipefail
 
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 724ee09a472..663d7d25d53 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -18,6 +18,7 @@ jobs:
     with:
       repository: pytorch/vision
       upload-artifact: docs
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
@@ -85,6 +86,7 @@ jobs:
       repository: pytorch/vision
       download-artifact: docs
       ref: gh-pages
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index 917bc54c832..23825fec589 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -14,6 +14,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
       repository: pytorch/vision
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
@@ -40,6 +41,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
       repository: pytorch/vision
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
@@ -73,6 +75,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
       repository: pytorch/vision
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 7c1a334e108..de8b9ffbf65 100644
--- a/.github/workflows/tests.yml
+++ b/.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: release/2.1
       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: release/2.1
       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: release/2.1
       script: |
         set -euxo pipefail
 
@@ -109,6 +112,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
       repository: pytorch/vision
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
@@ -139,6 +143,7 @@ jobs:
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@release/2.1
     with:
       repository: pytorch/vision
+      test-infra-ref: release/2.1
       script: |
         set -euo pipefail
 
diff --git a/.github/workflows/update-viablestrict.yml b/.github/workflows/update-viablestrict.yml
index ac81f3ff155..2c8583304cd 100644
--- a/.github/workflows/update-viablestrict.yml
+++ b/.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: release/2.1
     secrets:
       ROCKSET_API_KEY: ${{ secrets.ROCKSET_API_KEY }}
       GITHUB_DEPLOY_KEY : ${{ secrets.VISION_GITHUB_DEPLOY_KEY }}