From 3d9d40efdeea7011dc3c496ad9d55cfdc90aff92 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Lucchesi?= Date: Tue, 12 Aug 2025 16:30:17 +0200 Subject: [PATCH] [Bugfix][CI] Fix `test_remote_decode_lifecycle.py::test_short_prompt_lifecycle` (#22727) Signed-off-by: NickLucche --- tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py b/tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py index 1bddfef0f2ea..2f8228864e7b 100644 --- a/tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py +++ b/tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py @@ -130,8 +130,9 @@ def test_short_prompt_lifecycle(): # Confirm we do not have any memory leaks after req lifecycle. # We need to mark sending finish to clear data for persistent batch. scheduler_output = scheduler.schedule() - model_runner_output = copy.deepcopy(EMPTY_MODEL_RUNNER_OUTPUT) - model_runner_output.finished_sending = [request.request_id] + # Use create_model_runner_output to pass kv_connector_output along + model_runner_output = create_model_runner_output( + reqs=[request], finished_sending=[request.request_id]) scheduler.update_from_output(scheduler_output, model_runner_output) assert_scheduler_empty(scheduler)