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

[Chore] Separate out vllm.utils.collections (#26990)

Browse Source 
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2025-10-16 16:35:35 +08:00
committed by GitHub
parent 17838e50ef
commit d2740fafbf
29 changed files with 218 additions and 184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/utils_/test_utils.py
View File

@@ -38,7 +38,6 @@ from vllm.utils import (
sha256,
split_host_port,
split_zmq_path,
swap_dict_values,
unique_filepath,
)
@@ -516,30 +515,6 @@ def test_placeholder_module_error_handling():
_ = placeholder_attr.module
@pytest.mark.parametrize(
"obj,key1,key2",
[
# Tests for both keys exist
({1: "a", 2: "b"}, 1, 2),
# Tests for one key does not exist
({1: "a", 2: "b"}, 1, 3),
# Tests for both keys do not exist
({1: "a", 2: "b"}, 3, 4),
],
)
def test_swap_dict_values(obj, key1, key2):
original_obj = obj.copy()
swap_dict_values(obj, key1, key2)
if key1 in original_obj:
assert obj[key2] == original_obj[key1]
else:
assert key2 not in obj
if key2 in original_obj:
assert obj[key1] == original_obj[key2]
else:
assert key1 not in obj
def test_model_specification(
parser_with_config, cli_config_file, cli_config_file_with_model
):