From 6cb031a2a90f43ac6a91109b2c3d203b5b6946e3 Mon Sep 17 00:00:00 2001
From: Chukwuma Nwaugha <nwaughac@gmail.com>
Date: Sun, 14 Dec 2025 13:03:25 +0000
Subject: [PATCH] Simplify type assertion for output in VllmRunner

Signed-off-by: Chukwuma Nwaugha <nwaughac@gmail.com>
---
 tests/conftest.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 5b44847168396..17dc87b861ed8 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1018,8 +1018,7 @@ class VllmRunner:
 
         perplexities = []
         for output in outputs:
-            assert isinstance(output, TokensTextLogprobsPromptLogprobs)
-
+            output = cast(TokensTextLogprobsPromptLogprobs, output)
             token_datas = cast(list[LogprobsOnePosition | None], output[3])
             assert token_datas[0] is None
             token_log_probs = []