fix(CI): checkout head ref

Signed-off-by: Cameron Smith <[email protected]>

.github/workflows/CI.yaml

@@ -105,6 +105,8 @@ jobs:
     steps:
     - name: Checkout code
       uses: actions/checkout@v4
+      with:
+        ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.ref || github.ref }}
 
     - name: Create Flyte config from YAML template
       id: yq-process
@@ -149,6 +151,8 @@ jobs:
         echo "MODE=${{ needs.set-variables.outputs.mode }}"
     - name: Checkout code
       uses: actions/checkout@v4
+      with:
+        ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.ref || github.ref }}
 
     - name: Setup environment
       uses: ./.github/actions/setup_environment
@@ -192,6 +196,8 @@ jobs:
       contents: read
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.ref || github.ref }}
       - uses: DeterminateSystems/nix-installer-action@v9
       - uses: DeterminateSystems/magic-nix-cache-action@v2
       - run: nix flake check --impure --accept-flake-config
@@ -225,6 +231,8 @@ jobs:
       url: https://ghcr.io/sciexp/flytezendev
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.ref || github.ref }}
       - uses: docker/setup-qemu-action@v3
         with:
           platforms: arm64
@@ -259,8 +267,7 @@ jobs:
           NIX_IMAGE_SYSTEMS: "x86_64-linux" 
 
   execute-workflow:
-    # needs: [config-workflows, build-nix-images, set-variables]
-    needs: [config-workflows, set-variables]
+    needs: [config-workflows, build-nix-images, set-variables]
     if: ${{ needs.set-variables.outputs.skip_ci != 'true' && (contains(github.event.pull_request.labels.*.name, 'execute-workflow') || (github.event_name == 'workflow_dispatch' && inputs.run_execute_workflow)) }}
     runs-on: ubuntu-latest
     concurrency:
@@ -293,7 +300,7 @@ jobs:
         path: .flyte/
 
     - name: Setup tmate debug session
-      # if: ${{ inputs.debug_enabled == 'true' }}
+      if: ${{ inputs.debug_enabled == 'true' }}
       uses: mxschmitt/action-tmate@v3
 
     - name: Execute workflow