From a707b2054db7bb8b07a3128a55cd4857b20bb404 Mon Sep 17 00:00:00 2001 From: zijiexia <37504505+zijiexia@users.noreply.github.com> Date: Wed, 24 Jun 2026 00:56:52 -0700 Subject: [PATCH] [CI] Fix pre-commit failures in MLX backend tests (#29141) Co-authored-by: Claude Opus 4.8 --- .../unit/hardware_backend/mlx/test_attention_patching.py | 7 ++----- .../hardware_backend/mlx/test_mlx_runner_pool_contract.py | 2 +- 2 files changed, 3 insertions(+), 6 deletions(-) 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 d890564de..6f77b20c9 100644 --- a/test/registered/unit/hardware_backend/mlx/test_attention_patching.py +++ b/test/registered/unit/hardware_backend/mlx/test_attention_patching.py @@ -1207,8 +1207,7 @@ class TestMlxOverlapScheduler(unittest.TestCase): "_maybe_collect_customized_info": lambda *a, **k: None, } saved = { - name: getattr(SchedulerBatchResultProcessor, name) - for name in noop_stubs + name: getattr(SchedulerBatchResultProcessor, name) for name in noop_stubs } for name, value in noop_stubs.items(): setattr(SchedulerBatchResultProcessor, name, value) @@ -1229,9 +1228,7 @@ class TestMlxOverlapScheduler(unittest.TestCase): logits_output = SimpleNamespace(customized_info=None) original_release = batch_result_processor_module.release_kv_cache original_get_indexer = batch_result_processor_module.get_global_indexer_capturer - original_get_server_args = ( - batch_result_processor_module.get_global_server_args - ) + original_get_server_args = batch_result_processor_module.get_global_server_args def fake_release_kv_cache(release_req, tree_cache, is_insert=False): events.append(("release", release_req.rid)) diff --git a/test/registered/unit/hardware_backend/mlx/test_mlx_runner_pool_contract.py b/test/registered/unit/hardware_backend/mlx/test_mlx_runner_pool_contract.py index e73162631..266fdd446 100644 --- a/test/registered/unit/hardware_backend/mlx/test_mlx_runner_pool_contract.py +++ b/test/registered/unit/hardware_backend/mlx/test_mlx_runner_pool_contract.py @@ -76,4 +76,4 @@ class TestMlxRunnerPoolContract(unittest.TestCase): if __name__ == "__main__": - unittest.main() \ No newline at end of file + unittest.main()