From 533d1229ce348bfe442037d7a3ea7a3df10ddb1e Mon Sep 17 00:00:00 2001
From: Matthias Reso <13337103+mreso@users.noreply.github.com>
Date: Wed, 21 Aug 2024 18:27:46 +0000
Subject: [PATCH] Use env.FORK for CI like #3266 did

---
 .github/workflows/docker-ci.yaml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/docker-ci.yaml b/.github/workflows/docker-ci.yaml
index cd6d6feb70..37d9e6e45d 100644
--- 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