Move LogprobsOnePosition import to TYPE_CHECKING block in conftest.py

Signed-off-by: Chukwuma Nwaugha <nwaughac@gmail.com>
This commit is contained in:
Chukwuma Nwaugha 2025-12-17 16:29:01 +00:00
parent 76802a010b
commit 101c957bff

View File

@ -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