From fd7cd4b883e7720fbbfbc0ca6aa9404bf2f8e3f1 Mon Sep 17 00:00:00 2001 From: "Kevin H. Luu" Date: Sat, 12 Sep 2026 11:29:58 -0700 Subject: [PATCH] [CI] Fix Qwen3 Omni DSpark load test config (#56600) Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: OpenAI Codex --- tests/model_executor/test_qwen3_omni.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/model_executor/test_qwen3_omni.py b/tests/model_executor/test_qwen3_omni.py index e507b82d9c..92d527e625 100644 --- a/tests/model_executor/test_qwen3_omni.py +++ b/tests/model_executor/test_qwen3_omni.py @@ -354,6 +354,7 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary(): parallel_config=SimpleNamespace(tensor_parallel_size=1), attention_config=SimpleNamespace(backend=None), cache_config=SimpleNamespace(), + load_config=SimpleNamespace(), model_config=SimpleNamespace(get_vocab_size=Mock(return_value=100)), ) @@ -368,6 +369,10 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary(): "vllm.v1.worker.gpu.spec_decode.eagle.utils.get_pp_group", return_value=SimpleNamespace(world_size=1), ), + patch( + "vllm.v1.worker.gpu.spec_decode.utils.get_pp_group", + return_value=SimpleNamespace(world_size=1), + ), patch( "vllm.compilation.backends.set_model_tag", return_value=nullcontext(),