From 6256936d09081ecbc649ea9a4dd7307213af5316 Mon Sep 17 00:00:00 2001
From: Alison Shao <54658187+alisonshao@users.noreply.github.com>
Date: Fri, 2 Jan 2026 17:58:11 -0800
Subject: [PATCH] Fix: Allow build-wheels to run when target_stage is set
 (#16324)

---
 .github/workflows/pr-test.yml | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 8f8bf5c1f..242d64591 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -177,9 +177,7 @@ jobs:
 
   sgl-kernel-build-wheels:
     needs: [check-changes, call-gate]
-    if: |
-      !inputs.target_stage &&
-      needs.check-changes.outputs.sgl_kernel == 'true'
+    if: needs.check-changes.outputs.sgl_kernel == 'true'
     runs-on: x64-kernel-build-node
     strategy:
       matrix:
@@ -220,9 +218,7 @@ jobs:
 
   sgl-kernel-build-wheels-arm:
     needs: [check-changes, call-gate]
-    if: |
-      !inputs.target_stage &&
-      needs.check-changes.outputs.sgl_kernel == 'true'
+    if: needs.check-changes.outputs.sgl_kernel == 'true'
     runs-on: arm-kernel-build-node
     strategy:
       matrix: