biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Convert formatting to use ruff instead of yapf + isort (#26247)

Browse Source 
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-10-05 15:06:22 +01:00
committed by GitHub
parent 17edd8a807
commit d6953beb91
1508 changed files with 115244 additions and 94146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/utils_/test_tensor_schema.py
View File

@@ -6,8 +6,7 @@ import torch
from vllm.model_executor.models.glm4_1v import Glm4vImageEmbeddingInputs
from vllm.model_executor.models.granite_speech import GraniteSpeechAudioInputs
from vllm.model_executor.models.hyperclovax_vision import (
HCXVisionVideoPixelInputs)
from vllm.model_executor.models.hyperclovax_vision import HCXVisionVideoPixelInputs
from vllm.model_executor.models.phi3v import Phi3VImagePixelInputs
@@ -56,9 +55,10 @@ def test_tensor_schema_rank_mismatch():
def test_tensor_schema_missing_required_field():
with pytest.raises(ValueError,
match="Required field 'pixel_values' is missing"):
Phi3VImagePixelInputs(image_sizes=torch.randint(0, 256, (16, 2)), )
with pytest.raises(ValueError, match="Required field 'pixel_values' is missing"):
Phi3VImagePixelInputs(
image_sizes=torch.randint(0, 256, (16, 2)),
)
def test_tensor_schema_symbolic_dim_mismatch():
@@ -140,10 +140,7 @@ def test_tensor_schema_with_valid_resolve_binding_dims():
Phi3VImagePixelInputs(
pixel_values=pixel_values,
image_sizes=image_sizes,
resolve_bindings={
"h": 336,
"w": 336
},
resolve_bindings={"h": 336, "w": 336},
)
@@ -156,10 +153,7 @@ def test_tensor_schema_with_invalid_resolve_binding_dims():
Phi3VImagePixelInputs(
pixel_values=pixel_values,
image_sizes=image_sizes,
resolve_bindings={
"h": 336,
"w": 336
},
resolve_bindings={"h": 336, "w": 336},
)