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