From f408d05c523c25e2f638a13cb34a2dab3dcb2754 Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Mon, 25 Mar 2024 11:55:46 -0700 Subject: [PATCH] hotfix isort on logprobs ranks pr (#3622) --- tests/samplers/test_ranks.py | 1 + tests/samplers/test_sampler.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/samplers/test_ranks.py b/tests/samplers/test_ranks.py index 7f6f1c00..5e93238d 100644 --- a/tests/samplers/test_ranks.py +++ b/tests/samplers/test_ranks.py @@ -1,4 +1,5 @@ import pytest + from vllm import SamplingParams MODELS = ["facebook/opt-125m"] diff --git a/tests/samplers/test_sampler.py b/tests/samplers/test_sampler.py index 65b14fa3..1626b722 100644 --- a/tests/samplers/test_sampler.py +++ b/tests/samplers/test_sampler.py @@ -9,8 +9,8 @@ from transformers import GenerationConfig, GenerationMixin from vllm.model_executor.layers.sampler import Sampler from vllm.model_executor.utils import set_random_seed from vllm.sequence import SamplingParams, SequenceData, SequenceGroupMetadata -from vllm.worker.model_runner import ModelRunner from vllm.utils import Counter +from vllm.worker.model_runner import ModelRunner class MockLogitsSampler(Sampler):