From b1bb0de09f176d4d8a139ab2250ab2f38e60f953 Mon Sep 17 00:00:00 2001 From: Chukwuma Nwaugha Date: Thu, 11 Dec 2025 20:49:05 +0000 Subject: [PATCH] cleanup Signed-off-by: Chukwuma Nwaugha --- tests/conftest.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/conftest.py b/tests/conftest.py index d8ffa8b34f8dd..19ba271ff6ed5 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -992,9 +992,11 @@ class VllmRunner: perplexities = [] for output in outputs: assert isinstance(output, TokensTextLogprobsPromptLogprobs) + token_datas = cast(list[LogprobsOnePosition | None], output[3]) assert token_datas[0] is None token_log_probs = [] + for token_data in token_datas[1:]: assert token_data is not None assert len(token_data) == 1