From 2ab96d0fe0b156604a3dac29cdac739adb5e8e5a Mon Sep 17 00:00:00 2001 From: Ke Bao Date: Mon, 10 Aug 2026 00:55:55 +0800 Subject: [PATCH] Fix mlx unit test batch mock (#34181) --- .../unit/hardware_backend/mlx/test_attention_patching.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py index 36d874d02..8ce22bedd 100644 --- a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py +++ b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py @@ -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)