From 2777c31d93e79fa5320e815b6b2e8b4f5ba918fa Mon Sep 17 00:00:00 2001 From: Jing Wang Date: Wed, 9 Sep 2026 13:05:03 +0800 Subject: [PATCH] Fix Qwen3 Omni DSpark loader mock Signed-off-by: Jing Wang --- tests/model_executor/test_qwen3_omni.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/model_executor/test_qwen3_omni.py b/tests/model_executor/test_qwen3_omni.py index dea87a122694..e507b82d9c5e 100644 --- a/tests/model_executor/test_qwen3_omni.py +++ b/tests/model_executor/test_qwen3_omni.py @@ -347,10 +347,11 @@ def test_dspark_shares_target_embedding_with_smaller_draft_vocabulary(): vllm_config = SimpleNamespace( speculative_config=SimpleNamespace( draft_model_config=draft_model_config, - draft_parallel_config=SimpleNamespace(), + draft_parallel_config=SimpleNamespace(tensor_parallel_size=1), attention_backend=None, kv_cache_dtype=None, ), + parallel_config=SimpleNamespace(tensor_parallel_size=1), attention_config=SimpleNamespace(backend=None), cache_config=SimpleNamespace(), model_config=SimpleNamespace(get_vocab_size=Mock(return_value=100)),