From 00f088f6beae5191c5e509033155b092363a1821 Mon Sep 17 00:00:00 2001
From: Cheng Wan <54331508+ch-wan@users.noreply.github.com>
Date: Sat, 4 Jul 2026 14:23:40 -0700
Subject: [PATCH] [fix] Wrap the sp_shard test entry point in sys.exit so
 failures propagate (#30138)

---
 python/sglang/multimodal_gen/test/unit/test_sp_shard.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/sglang/multimodal_gen/test/unit/test_sp_shard.py b/python/sglang/multimodal_gen/test/unit/test_sp_shard.py
index f16173426..4f7e8261b 100644
--- a/python/sglang/multimodal_gen/test/unit/test_sp_shard.py
+++ b/python/sglang/multimodal_gen/test/unit/test_sp_shard.py
@@ -1,5 +1,7 @@
 """Unit tests for the unified SP shard helpers (pure logic, no distributed)."""
 
+import sys
+
 import pytest
 import torch
 
@@ -210,4 +212,4 @@ def test_gather_seq_trims(monkeypatch):
 
 
 if __name__ == "__main__":
-    pytest.main([__file__, "-q"])
+    sys.exit(pytest.main([__file__, "-q"]))