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(),