Fix mlx unit test batch mock (#34181)

2026-08-10 00:55:55 +08:00
parent fb72a37fde
commit 2ab96d0fe0
1 changed files with 5 additions and 1 deletions
@@ -1222,7 +1222,11 @@ class TestMlxOverlapScheduler(unittest.TestCase):
                set_completion_time=lambda: events.append(("completion", "r0"))
            ),
        )
-        batch = SimpleNamespace()
+        batch = SimpleNamespace(
+            mamba_track_mask_cpu=None,
+            mamba_track_mask_next_cpu=None,
+            mamba_decode_batch_idx_cpu=None,
+        )
        result = SimpleNamespace()
        i = 0
        logits_output = SimpleNamespace(customized_info=None)