fix unit tests and add deepseek test case

Signed-off-by: zixi-qi <qizixi@meta.com>

Author: zixi-qi

tests/v1/e2e/test_spec_decode.py

@@ -229,8 +229,9 @@ def test_eagle_correctness(
 
 @pytest.mark.parametrize(["model_setup", "mm_enabled"], [
     (("mtp", "XiaomiMiMo/MiMo-7B-Base", 1), False),
+    (("mtp", "ZixiQi/DeepSeek-V3-4layers-MTP-FP8", 1), False),
 ],
-                         ids=["mtp"])
+                         ids=["mimo", "deepseek"])
 @pytest.mark.parametrize("attn_backend",
                          get_attn_backend_list_based_on_platform())
 def test_mtp_correctness(

tests/v1/spec_decode/test_mtp.py

@@ -201,10 +201,7 @@ def create_deterministic_logits(batch_size, vocab_size, token_offset):
     )
 
     proposer.runner = mock.MagicMock()
-    proposer.runner.attn_groups.append([mock.MagicMock()])
-    proposer.runner.attn_groups[0][0].metadata_builders = [
-        attn_metadata_builder
-    ]
+    proposer.attn_metadata_builder = attn_metadata_builder
 
     # Run propose
     result = proposer.propose(target_token_ids=target_token_ids,