[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 <codex@openai.com>
Kevin H. Luu committed
Sep 12, 2026 at 11:29 UTC
fd7cd4b883e7720fbbfbc0ca6aa9404bf2f8e3f1
1 file changed
+5
tests/model_executor/test_qwen3_omni.py
+5
@@ -354,6 +354,7 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary():
354
parallel_config=SimpleNamespace(tensor_parallel_size=1),
355
attention_config=SimpleNamespace(backend=None),
356
cache_config=SimpleNamespace(),
357
+ load_config=SimpleNamespace(),
358
model_config=SimpleNamespace(get_vocab_size=Mock(return_value=100)),
359
)
360
@@ -368,6 +369,10 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary():
369
"vllm.v1.worker.gpu.spec_decode.eagle.utils.get_pp_group",
370
return_value=SimpleNamespace(world_size=1),
371
),
372
+ patch(
373
+ "vllm.v1.worker.gpu.spec_decode.utils.get_pp_group",
374
+ return_value=SimpleNamespace(world_size=1),
375
+ ),
376
patch(
377
"vllm.compilation.backends.set_model_tag",
378
return_value=nullcontext(),