From b05ae5dc008850a620dec6de66635dec2b5913fd Mon Sep 17 00:00:00 2001 From: "Kevin H. Luu" Date: Tue, 18 Aug 2026 18:04:10 -0700 Subject: [PATCH] [CI][Bugfix] Complete DeepSeek-V4 FSE test fixture contract (#52842) Signed-off-by: Thomas Parnell Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: Thomas Parnell Co-authored-by: Claude --- tests/model_executor/layers/test_fused_shared_expert.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/model_executor/layers/test_fused_shared_expert.py b/tests/model_executor/layers/test_fused_shared_expert.py index 7080c66f12..c667f326f6 100644 --- a/tests/model_executor/layers/test_fused_shared_expert.py +++ b/tests/model_executor/layers/test_fused_shared_expert.py @@ -479,6 +479,8 @@ def test_models_fse_init( is_mm_prefix_lm=False, multimodal_config=None, quantization_config=None, + runner_type="generate", + is_moe=True, ) vllm_config.parallel_config.enable_expert_parallel = False if model_type == "deepseek_v4": @@ -535,7 +537,8 @@ def test_models_fse_init( ) vllm_config.speculative_config = SimpleNamespace( - draft_model_config=SimpleNamespace(hf_config=config) + draft_model_config=SimpleNamespace(hf_config=config), + method="mtp", ) mtp = DeepSeekV4MTP(vllm_config=vllm_config) assert model.is_fused_shared_expert_enabled is (fse_enabled and not exclude)