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(