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"]))