xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-11 01:35:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix][CI] Fix test_remote_decode_lifecycle.py::test_short_prompt_lifecycle (#22727)

Browse Source 
Signed-off-by: NickLucche <nlucches@redhat.com>
This commit is contained in:
Nicolò Lucchesi 2025-08-12 16:30:17 +02:00 committed by GitHub
parent 67c153b88a
commit 3d9d40efde
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/v1/kv_connector/unit/test_remote_decode_lifecycle.py
View File

@ -130,8 +130,9 @@ def test_short_prompt_lifecycle():
# Confirm we do not have any memory leaks after req lifecycle. # Confirm we do not have any memory leaks after req lifecycle.
# We need to mark sending finish to clear data for persistent batch. # We need to mark sending finish to clear data for persistent batch.
scheduler_output = scheduler.schedule() scheduler_output = scheduler.schedule()
model_runner_output = copy.deepcopy(EMPTY_MODEL_RUNNER_OUTPUT) # Use create_model_runner_output to pass kv_connector_output along
model_runner_output.finished_sending = [request.request_id] model_runner_output = create_model_runner_output(
reqs=[request], finished_sending=[request.request_id])
scheduler.update_from_output(scheduler_output, model_runner_output) scheduler.update_from_output(scheduler_output, model_runner_output)
assert_scheduler_empty(scheduler) assert_scheduler_empty(scheduler)