A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 14, 05:30 UTC
[Misc] Move tensor schema tests (#22612)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
Cyrus Leung committed
Aug 11, 2025 at 15:15 UTC
ebf7605b0dd58ff5d572d1918e52ca732025eee0
5 files changed
+12
-6
.buildkite/test-pipeline.yaml
+4
-3
@@ -57,9 +57,10 @@ steps:
57
- vllm/
58
- tests/mq_llm_engine
59
- tests/async_engine
60
- - tests/test_inputs
60
+ - tests/test_inputs.py
61
+ - tests/test_outputs.py
62
- tests/multimodal
62
- - tests/test_utils
63
+ - tests/utils_
64
- tests/worker
65
- tests/standalone_tests/lazy_imports.py
66
commands:
@@ -70,7 +71,7 @@ steps:
71
- pytest -v -s test_inputs.py
72
- pytest -v -s test_outputs.py
73
- pytest -v -s multimodal
73
- - pytest -v -s test_utils.py # Utils
74
+ - pytest -v -s utils_ # Utils
75
- pytest -v -s worker # Worker
76
77
- label: Python-only Installation Test
tests/utils_/__init__.py
new
+6
@@ -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/utils_/test_tensor_schema.py
renamed
tests/utils_/test_utils.py
renamed
+1
-2
@@ -5,7 +5,6 @@
5
import asyncio
6
import hashlib
7
import json
8
-import logging
8
import pickle
9
import socket
10
from collections.abc import AsyncIterator
@@ -29,7 +28,7 @@ from vllm.utils import (CacheInfo, FlexibleArgumentParser, LRUCache,
28
merge_async_iterators, sha256, split_host_port,
29
split_zmq_path, supports_kw, swap_dict_values)
30
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
32
33
34
@pytest.mark.asyncio
tools/check_pickle_imports.py
+1
-1
@@ -32,7 +32,7 @@ ALLOWED_FILES = set([
32
'vllm/multimodal/hasher.py',
33
'vllm/transformers_utils/config.py',
34
'vllm/model_executor/models/registry.py',
35
- 'tests/test_utils.py',
35
+ 'tests/utils_/test_utils.py',
36
'tests/tokenization/test_cached_tokenizer.py',
37
'vllm/distributed/utils.py',
38
'vllm/distributed/parallel_state.py',