From 2263f8a3de64f4cf16488fb43369714c736612a0 Mon Sep 17 00:00:00 2001 From: Nick Hill Date: Thu, 11 Jun 2026 20:26:17 -0700 Subject: [PATCH] [CI][BugFix] Fix broken `test_mamba_prefix_cache.py` due to stale mock (#45345) Signed-off-by: Nick Hill --- tests/v1/e2e/general/test_mamba_prefix_cache.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/v1/e2e/general/test_mamba_prefix_cache.py b/tests/v1/e2e/general/test_mamba_prefix_cache.py index ceae041c6f..e857b12728 100644 --- a/tests/v1/e2e/general/test_mamba_prefix_cache.py +++ b/tests/v1/e2e/general/test_mamba_prefix_cache.py @@ -181,6 +181,7 @@ def get_fake_allocate_slots_fn(original_allocate_slots_fn: Callable): num_encoder_tokens: int = 0, full_sequence_must_fit: bool = False, reserved_blocks: int = 0, + has_scheduled_reqs: bool = True, ): ret = original_allocate_slots_fn( self, @@ -194,6 +195,7 @@ def get_fake_allocate_slots_fn(original_allocate_slots_fn: Callable): num_encoder_tokens, full_sequence_must_fit, reserved_blocks, + has_scheduled_reqs, ) if cur_step_action is not None: cur_block_ids = self.coordinator.single_type_managers[0].req_to_blocks[