Testing pytorch/test-infra#6013

Author: huydhn

.github/workflows/_linux-build.yml

@@ -136,7 +136,7 @@ jobs:
 
       - name: Calculate docker image
         id: calculate-docker-image
-        uses: pytorch/test-infra/.github/actions/calculate-docker-image@main
+        uses: pytorch/test-infra/.github/actions/calculate-docker-image@wait-for-docker-build
         if: inputs.build-environment != 'linux-s390x-binary-manywheel'
         with:
           docker-image-name: ${{ inputs.docker-image-name }}

.github/workflows/docker-builds.yml

@@ -104,7 +104,7 @@ jobs:
 
       - name: Build docker image
         id: build-docker-image
-        uses: pytorch/test-infra/.github/actions/calculate-docker-image@main
+        uses: pytorch/test-infra/.github/actions/calculate-docker-image@wait-for-docker-build
         with:
           docker-image-name: ${{ matrix.docker-image-name }}
           always-rebuild: true