From c2b55de5b9ea618cd296665c31c1ffb82b366900 Mon Sep 17 00:00:00 2001 From: Zhou Taichang Date: Tue, 28 Jul 2026 23:25:57 +0800 Subject: [PATCH] [Test] Give the logprobs scheduler doubles the vLLM 0.26 contracts (#5483) Signed-off-by: Gao Han Co-authored-by: tzhouam Co-authored-by: Claude Fable 5 Co-authored-by: Gao Han --- tests/core/sched/test_omni_ar_scheduler_logprobs.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/core/sched/test_omni_ar_scheduler_logprobs.py b/tests/core/sched/test_omni_ar_scheduler_logprobs.py index 130ac1c88..394063843 100644 --- a/tests/core/sched/test_omni_ar_scheduler_logprobs.py +++ b/tests/core/sched/test_omni_ar_scheduler_logprobs.py @@ -76,6 +76,7 @@ class _Request: self.status = RequestStatus.RUNNING self.sampling_params = SimpleNamespace(num_logprobs=1) self.num_computed_tokens = 0 + self.num_in_flight_tokens = 0 self.num_output_placeholders = 0 self.has_encoder_inputs = False self.pooling_params = None @@ -104,7 +105,6 @@ class _RequestQueue(list): self.remove(request) -@pytest.mark.skip(reason="#issue 5484") def test_invalid_logprobs_finish_only_the_affected_scheduler_request() -> None: """A bad runner response must not bring down unrelated batch requests.""" bad = _Request("bad") @@ -138,7 +138,8 @@ def test_invalid_logprobs_finish_only_the_affected_scheduler_request() -> None: scheduler.requests.pop(request.request_id, None) scheduler.finished_req_ids.add(request.request_id) scheduler.finished_req_ids_dict[request.client_index].add(request.request_id) - return None + # vLLM 0.26 contract: (kv_xfer_params, ec_xfer_params) + return None, None scheduler._update_request_with_output = update_request scheduler._process_kv_transfer_trigger = lambda _request, _tokens: False