Use env.FORK for CI like pytorch#3266 did

mreso · Aug 21, 2024 · 533d122 · 533d122
1 parent d1ed4f1
commit 533d122
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/docker-ci.yaml b/.github/workflows/docker-ci.yaml
@@ -39,7 +39,11 @@ jobs:
         working-directory: docker
         run: |
           IMAGE_TAG=test-image-${{ matrix.python-version }}
-          REPO_URL=$(echo ${{ github.repositoryUrl }} | sed 's/^git:\/\//https:\/\//')
+          if [[ -z "${{ env.FORK }}" ]]; then
+            REPO_URL="https://github.com/pytorch/serve.git"
+          else
+            REPO_URL="https://github.com/${{ env.FORK }}"
+          fi
           ./build_image.sh -py "${{ matrix.python-version }}" -t "${IMAGE_TAG}" -b "${{ steps.branch-name.outputs.GITHUB_BRANCH }}" -repo ${REPO_URL} -s
           echo "IMAGE_TAG=${IMAGE_TAG}" >> $GITHUB_OUTPUT