From a1033a9542a8f42e4d69e5ed95598c814956eb13 Mon Sep 17 00:00:00 2001 From: qli88 Date: Fri, 21 Nov 2025 15:58:03 +0000 Subject: [PATCH] ruff change Signed-off-by: qli88 --- tests/samplers/test_logprobs.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/samplers/test_logprobs.py b/tests/samplers/test_logprobs.py index 5ae84c7ef4c66..8bc43b1f2e03c 100644 --- a/tests/samplers/test_logprobs.py +++ b/tests/samplers/test_logprobs.py @@ -25,7 +25,9 @@ def test_ranks( flat_logprobs, example_prompts, ): - with vllm_runner(model, dtype=dtype, max_logprobs=MAX_LOGPROBS, enforce_eager=True) as vllm_model: + with vllm_runner( + model, dtype=dtype, max_logprobs=MAX_LOGPROBS, enforce_eager=True + ) as vllm_model: tokenizer = vllm_model.llm.get_tokenizer() example_prompt_tokens = [tokenizer.encode(prompt) for prompt in example_prompts] sampling_params = SamplingParams(