Skip to content

Commit f408d05

Browse files
authored
hotfix isort on logprobs ranks pr (#3622)
1 parent 0b4997e commit f408d05

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tests/samplers/test_ranks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import pytest
2+
23
from vllm import SamplingParams
34

45
MODELS = ["facebook/opt-125m"]

tests/samplers/test_sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@
99
from vllm.model_executor.layers.sampler import Sampler
1010
from vllm.model_executor.utils import set_random_seed
1111
from vllm.sequence import SamplingParams, SequenceData, SequenceGroupMetadata
12-
from vllm.worker.model_runner import ModelRunner
1312
from vllm.utils import Counter
13+
from vllm.worker.model_runner import ModelRunner
1414

1515

1616
class MockLogitsSampler(Sampler):

0 commit comments

Comments
 (0)