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

34
tests/test_sequence.py
View File

@@ -7,7 +7,6 @@ from vllm.sequence import IntermediateTensors
def test_sequence_intermediate_tensors_equal():
class AnotherIntermediateTensors(IntermediateTensors):
pass
@@ -20,22 +19,31 @@ def test_sequence_intermediate_tensors_equal():
assert empty_intermediate_tensors_1 == empty_intermediate_tensors_2
different_key_intermediate_tensors_1 = IntermediateTensors(
{"1": torch.zeros([2, 4], dtype=torch.int32)})
{"1": torch.zeros([2, 4], dtype=torch.int32)}
)
difference_key_intermediate_tensors_2 = IntermediateTensors(
{"2": torch.zeros([2, 4], dtype=torch.int32)})
assert (different_key_intermediate_tensors_1
!= difference_key_intermediate_tensors_2)
{"2": torch.zeros([2, 4], dtype=torch.int32)}
)
assert different_key_intermediate_tensors_1 != difference_key_intermediate_tensors_2
same_key_different_value_intermediate_tensors_1 = IntermediateTensors(
{"1": torch.zeros([2, 4], dtype=torch.int32)})
{"1": torch.zeros([2, 4], dtype=torch.int32)}
)
same_key_different_value_intermediate_tensors_2 = IntermediateTensors(
{"1": torch.zeros([2, 5], dtype=torch.int32)})
assert (same_key_different_value_intermediate_tensors_1
!= same_key_different_value_intermediate_tensors_2)
{"1": torch.zeros([2, 5], dtype=torch.int32)}
)
assert (
same_key_different_value_intermediate_tensors_1
!= same_key_different_value_intermediate_tensors_2
)
same_key_same_value_intermediate_tensors_1 = IntermediateTensors(
{"1": torch.zeros([2, 4], dtype=torch.int32)})
{"1": torch.zeros([2, 4], dtype=torch.int32)}
)
same_key_same_value_intermediate_tensors_2 = IntermediateTensors(
{"1": torch.zeros([2, 4], dtype=torch.int32)})
assert (same_key_same_value_intermediate_tensors_1 ==
same_key_same_value_intermediate_tensors_2)
{"1": torch.zeros([2, 4], dtype=torch.int32)}
)
assert (
same_key_same_value_intermediate_tensors_1
== same_key_same_value_intermediate_tensors_2
)