diff --git a/test/registered/unit/models/test_deepseek_v4_shared_expert_fusion.py b/test/registered/unit/models/test_deepseek_v4_shared_expert_fusion.py index 792691052..46be9817a 100644 --- a/test/registered/unit/models/test_deepseek_v4_shared_expert_fusion.py +++ b/test/registered/unit/models/test_deepseek_v4_shared_expert_fusion.py @@ -12,7 +12,7 @@ from sglang.srt.layers.moe.utils import ( ) from sglang.srt.models.deepseek_v4 import DeepseekV4ForCausalLM from sglang.srt.models.deepseek_v4_dspark import DeepseekV4ForCausalLMDSpark -from sglang.srt.runtime_context import get_context, get_exec, get_flags +from sglang.srt.runtime_context import get_context, get_exec, get_flags, get_parallel from sglang.test.ci.ci_register import register_cpu_ci from sglang.test.test_utils import CustomTestCase @@ -27,6 +27,12 @@ class TestDeepseekV4SharedExpertFusionPolicy(CustomTestCase): on the ACTIVE moe flag, and the config bag keeps the user's intent. """ + def setUp(self): + super().setUp() + cm = get_parallel().override(moe_ep_size=1) + cm.__enter__() + self.addCleanup(cm.__exit__, None, None, None) + def _publish(self, enforce): override = get_context().override_server_args( enforce_shared_experts_fusion=enforce