[CI][Bugfix] Complete DeepSeek-V4 FSE test fixture contract (#52842)
Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com> Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: Thomas Parnell <tpa@zurich.ibm.com> Co-authored-by: Claude <noreply@anthropic.com>
Kevin H. Luu committed
Aug 18, 2026 at 18:04 UTC
b05ae5dc008850a620dec6de66635dec2b5913fd
1 file changed
+4
-1
tests/model_executor/layers/test_fused_shared_expert.py
+4
-1
@@ -479,6 +479,8 @@ def test_models_fse_init(
479
is_mm_prefix_lm=False,
480
multimodal_config=None,
481
quantization_config=None,
482
+ runner_type="generate",
483
+ is_moe=True,
484
)
485
vllm_config.parallel_config.enable_expert_parallel = False
486
if model_type == "deepseek_v4":
@@ -535,7 +537,8 @@ def test_models_fse_init(
537
)
538
539
vllm_config.speculative_config = SimpleNamespace(
538
- draft_model_config=SimpleNamespace(hf_config=config)
540
+ draft_model_config=SimpleNamespace(hf_config=config),
541
+ method="mtp",
542
)
543
mtp = DeepSeekV4MTP(vllm_config=vllm_config)
544
assert model.is_fused_shared_expert_enabled is (fse_enabled and not exclude)