hotfix isort on logprobs ranks pr (#3622)

IBM · Mar 25, 2024 · 92eefa2 · 92eefa2
1 parent dabe1a1
commit 92eefa2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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):