xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-09 23:29:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Simplify type assertion for output in VllmRunner

Browse Source 
Signed-off-by: Chukwuma Nwaugha <nwaughac@gmail.com>
This commit is contained in:
Chukwuma Nwaugha 2025-12-14 13:03:25 +00:00
parent 4f2c8dab9a
commit 6cb031a2a9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/conftest.py
View File

@ -1018,8 +1018,7 @@ class VllmRunner:
perplexities = [] perplexities = []
for output in outputs: for output in outputs:
assert isinstance(output, TokensTextLogprobsPromptLogprobs) output = cast(TokensTextLogprobsPromptLogprobs, output)
token_datas = cast(list[LogprobsOnePosition | None], output[3]) token_datas = cast(list[LogprobsOnePosition | None], output[3])
assert token_datas[0] is None assert token_datas[0] is None
token_log_probs = [] token_log_probs = []