Skip to content

Commit 22cae24

Browse files
DarkLight1337jingyu
authored andcommitted
[Misc] Move tensor schema tests (vllm-project#22612)
Signed-off-by: DarkLight1337 <[email protected]> Signed-off-by: jingyu <[email protected]>
1 parent b4c9c69 commit 22cae24

File tree

5 files changed

+12
-6
lines changed

5 files changed

+12
-6
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,10 @@ steps:
5757
- vllm/
5858
- tests/mq_llm_engine
5959
- tests/async_engine
60-
- tests/test_inputs
60+
- tests/test_inputs.py
61+
- tests/test_outputs.py
6162
- tests/multimodal
62-
- tests/test_utils
63+
- tests/utils_
6364
- tests/worker
6465
- tests/standalone_tests/lazy_imports.py
6566
commands:
@@ -70,7 +71,7 @@ steps:
7071
- pytest -v -s test_inputs.py
7172
- pytest -v -s test_outputs.py
7273
- pytest -v -s multimodal
73-
- pytest -v -s test_utils.py # Utils
74+
- pytest -v -s utils_ # Utils
7475
- pytest -v -s worker # Worker
7576

7677
- label: Python-only Installation Test

tests/utils_/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
# SPDX-License-Identifier: Apache-2.0
2+
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3+
"""
4+
This module is named `utils_` instead of `utils` to avoid obscuring
5+
`tests/utils.py`.
6+
"""

tests/test_utils.py renamed to tests/utils_/test_utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import asyncio
66
import hashlib
77
import json
8-
import logging
98
import pickle
109
import socket
1110
from collections.abc import AsyncIterator
@@ -29,7 +28,7 @@
2928
merge_async_iterators, sha256, split_host_port,
3029
split_zmq_path, supports_kw, swap_dict_values)
3130

32-
from .utils import create_new_process_for_each_test, error_on_warning
31+
from ..utils import create_new_process_for_each_test, error_on_warning
3332

3433

3534
@pytest.mark.asyncio

tools/check_pickle_imports.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
'vllm/multimodal/hasher.py',
3333
'vllm/transformers_utils/config.py',
3434
'vllm/model_executor/models/registry.py',
35-
'tests/test_utils.py',
35+
'tests/utils_/test_utils.py',
3636
'tests/tokenization/test_cached_tokenizer.py',
3737
'vllm/distributed/utils.py',
3838
'vllm/distributed/parallel_state.py',

0 commit comments

Comments
 (0)