[CI][BugFix] Fix broken test_mamba_prefix_cache.py due to stale mock (#45345)

Signed-off-by: Nick Hill <nickhill123@gmail.com>
2026-06-12 03:26:17 +00:00
parent c1076839c9
commit 2263f8a3de
1 changed files with 2 additions and 0 deletions
@@ -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[