[CI] Fix stale CP-v1 mock patches in NextN mm-embed test (#38280)

This commit is contained in:
Mohammad Miadh Angkad
2026-09-06 23:20:29 -07:00
committed by GitHub
parent 97dcbf9410
commit cfa989b7af
@@ -56,10 +56,6 @@ def _make_model_nextn(vocab_size: int, hidden_size: int):
model.rot_weight = None
model.alt_stream = None
model.quant_config = None
model.cp_rank = None
model.cp_size = None
model.dsa_enable_prefill_cp = False
model.mla_enable_prefill_cp = False
model.mtp_block = MagicMock(side_effect=lambda **kw: (kw["hidden_states"], None))
return model
@@ -96,9 +92,6 @@ class TestDeepseekNextNMmEmbed(CustomTestCase):
object.__setattr__(model, "embed_tokens", mock_embed)
with (
patch(
"sglang.srt.models.deepseek_nextn.is_cp_v2_active", return_value=False
),
patch(
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
return_value=False,
@@ -164,9 +157,6 @@ class TestDeepseekNextNMmEmbed(CustomTestCase):
embed_calls = mock_embed.call_args_list
with (
patch(
"sglang.srt.models.deepseek_nextn.is_cp_v2_active", return_value=False
),
patch(
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
return_value=False,