diff --git a/tests/samplers/test_ranks.py b/tests/samplers/test_ranks.py index 7f6f1c009315..5e93238d709e 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 65b14fa340ee..1626b7228207 100644 --- a/tests/samplers/test_sampler.py +++ b/tests/samplers/test_sampler.py @@ -9,8 +9,8 @@ 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):