CI Fixes pull request pull for gpu build (#29186)

thomasjpfan · web-flow · commit 1009ca42f505 · 2024-06-05T10:54:53.000+02:00
diff --git a/.github/workflows/cuda-gpu-ci.yml b/.github/workflows/cuda-gpu-ci.yml
@@ -18,14 +18,9 @@ jobs:
         with:
           python-version: '3.12'
       - name: Checkout main repository
-        uses: actions/checkout@v2
-      - name: Checkout a particular Pull Request
-        if: inputs.pr_id
-        env:
-          PR_ID: ${{ inputs.pr_id }}
-          GH_TOKEN: ${{ github.token }}
-        run: |
-          gh pr checkout ${{ env.PR_ID }}
+        uses: actions/checkout@v4
+        with:
+          ref: pull/${{ inputs.pr_id }}/head
       - name: Cache conda environment
         id: cache-conda
         uses: actions/cache@v3
