From 101c957bff1c1174b63f063b6383505f8862826b Mon Sep 17 00:00:00 2001 From: Chukwuma Nwaugha Date: Wed, 17 Dec 2025 16:29:01 +0000 Subject: [PATCH] Move LogprobsOnePosition import to TYPE_CHECKING block in conftest.py Signed-off-by: Chukwuma Nwaugha --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index 17dc87b861ed8..b571bf94e8bd8 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -58,7 +58,6 @@ from vllm.distributed import ( initialize_model_parallel, ) from vllm.logger import init_logger -from vllm.logprobs import LogprobsOnePosition from vllm.multimodal.utils import fetch_image from vllm.outputs import RequestOutput from vllm.sampling_params import BeamSearchParams @@ -70,6 +69,7 @@ from torch._inductor.utils import fresh_cache if TYPE_CHECKING: + from vllm.logprobs import LogprobsOnePosition from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast from transformers.generation.utils import GenerateOutput