Consolidate GitHub CLI installation steps as requested in PR review

Co-authored-by: leofang <5534781+leofang@users.noreply.github.com>

Author: Copilot

.github/workflows/test-wheel-linux.yml

@@ -149,12 +149,10 @@ jobs:
           SHA: ${{ github.sha }}
         run: ./ci/tools/env-vars test
 
-      - name: Download cuda-pathfinder build artifacts from main branch
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      - name: Install GitHub CLI
+        # gh is needed for artifact fetching in subsequent steps
         run: |
           # See https://github.com/cli/cli/blob/trunk/docs/install_linux.md#debian-ubuntu-linux-raspberry-pi-os-apt.
-          # gh is needed for artifact fetching.
           mkdir -p -m 755 /etc/apt/keyrings \
                 && out=$(mktemp) && wget -nv -O$out https://cli.github.com/packages/githubcli-archive-keyring.gpg \
                 && cat $out | tee /etc/apt/keyrings/githubcli-archive-keyring.gpg > /dev/null \
@@ -163,6 +161,10 @@ jobs:
           && apt update \
           && apt install gh -y
 
+      - name: Download cuda-pathfinder build artifacts from main branch
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
           MAIN_BRANCH="main"
           PATHFINDER_BASENAME="cuda-pathfinder-wheel"
           LATEST_MAIN_RUN_ID=$(gh run list -b ${MAIN_BRANCH} -L 1 -w "CI" -s completed -R NVIDIA/cuda-python --json databaseId | jq '.[]| .databaseId')
@@ -196,16 +198,6 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         run: |
-          # See https://github.com/cli/cli/blob/trunk/docs/install_linux.md#debian-ubuntu-linux-raspberry-pi-os-apt.
-          # gh is needed for artifact fetching.
-          mkdir -p -m 755 /etc/apt/keyrings \
-                && out=$(mktemp) && wget -nv -O$out https://cli.github.com/packages/githubcli-archive-keyring.gpg \
-                && cat $out | tee /etc/apt/keyrings/githubcli-archive-keyring.gpg > /dev/null \
-          && chmod go+r /etc/apt/keyrings/githubcli-archive-keyring.gpg \
-          && echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/githubcli-archive-keyring.gpg] https://cli.github.com/packages stable main" | tee /etc/apt/sources.list.d/github-cli.list > /dev/null \
-          && apt update \
-          && apt install gh -y
-
           OLD_BRANCH=$(cat .github/BACKPORT_BRANCH)
           OLD_BASENAME="cuda-bindings-python${PYTHON_VERSION_FORMATTED}-cuda*-${{ inputs.host-platform }}*"
           LATEST_PRIOR_RUN_ID=$(gh run list -b ${OLD_BRANCH} -L 1 -w "build-and-test.yml" -s completed -R NVIDIA/cuda-python --json databaseId | jq '.[]| .databaseId')

.github/workflows/test-wheel-windows.yml

@@ -117,25 +117,6 @@ jobs:
         shell: bash --noprofile --norc -xeuo pipefail {0}
         run: ./ci/tools/env-vars test
 
-      - name: Download cuda-pathfinder build artifacts from main branch
-        env:
-          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-        run: |
-          $MAIN_BRANCH = "main"
-          $PATHFINDER_BASENAME = "cuda-pathfinder-wheel"
-          $runData = gh run list -b $MAIN_BRANCH -L 1 -w "CI" -s completed -R NVIDIA/cuda-python --json databaseId | ConvertFrom-Json
-          if (-not $runData -or $runData.Length -eq 0 -or -not $runData[0].databaseId -or [string]::IsNullOrEmpty($runData[0].databaseId)) {
-              Write-Host "LATEST_MAIN_RUN_ID not found!"
-              exit 1
-          }
-          $LATEST_MAIN_RUN_ID = $runData[0].databaseId
-
-          gh run download $LATEST_MAIN_RUN_ID -p $PATHFINDER_BASENAME -R NVIDIA/cuda-python
-          Get-ChildItem -Path $PATHFINDER_BASENAME
-          New-Item -Path "./cuda_pathfinder" -ItemType Directory -Force
-          Move-Item -Path "$PATHFINDER_BASENAME/*.whl" -Destination "./cuda_pathfinder/"
-          Remove-Item -Path $PATHFINDER_BASENAME -Force
-
       - name: Download cuda-python build artifacts
         if: ${{ env.SKIP_CUDA_BINDINGS_TEST == '0'}}
         uses: actions/download-artifact@d3f86a106a0bac45b974a628896c90dbdf5c8093  # v4.3.0
@@ -165,6 +146,25 @@ jobs:
           }
           gh --version
 
+      - name: Download cuda-pathfinder build artifacts from main branch
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          $MAIN_BRANCH = "main"
+          $PATHFINDER_BASENAME = "cuda-pathfinder-wheel"
+          $runData = gh run list -b $MAIN_BRANCH -L 1 -w "CI" -s completed -R NVIDIA/cuda-python --json databaseId | ConvertFrom-Json
+          if (-not $runData -or $runData.Length -eq 0 -or -not $runData[0].databaseId -or [string]::IsNullOrEmpty($runData[0].databaseId)) {
+              Write-Host "LATEST_MAIN_RUN_ID not found!"
+              exit 1
+          }
+          $LATEST_MAIN_RUN_ID = $runData[0].databaseId
+
+          gh run download $LATEST_MAIN_RUN_ID -p $PATHFINDER_BASENAME -R NVIDIA/cuda-python
+          Get-ChildItem -Path $PATHFINDER_BASENAME
+          New-Item -Path "./cuda_pathfinder" -ItemType Directory -Force
+          Move-Item -Path "$PATHFINDER_BASENAME/*.whl" -Destination "./cuda_pathfinder/"
+          Remove-Item -Path $PATHFINDER_BASENAME -Force
+
       - name: Install zstd
         # the GPU runner image does not have zstd pre-installed... and it's needed by actions/cache
         if: ${{ matrix.LOCAL_CTK == '1' }}