Skip to content

Commit da40185

Browse files
committed
fix
Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>
1 parent 2e53427 commit da40185

File tree

1 file changed

+9
-11
lines changed

1 file changed

+9
-11
lines changed

tests/utils_/test_utils.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,16 +23,16 @@
2323
from vllm.config import ParallelConfig, VllmConfig, set_current_vllm_config
2424
from vllm.transformers_utils.detokenizer_utils import (
2525
convert_ids_list_to_tokens)
26-
from vllm.utils import (CacheInfo, FlexibleArgumentParser, LRUCache,
27-
MemorySnapshot, PlaceholderModule, StoreBoolean,
28-
bind_kv_cache, common_broadcastable_dtype,
29-
current_stream, deprecate_kwargs, get_open_port,
30-
get_tcp_uri, is_lossless_cast, join_host_port,
31-
make_zmq_path, make_zmq_socket, memory_profiling,
32-
merge_async_iterators, sha256, split_host_port,
33-
split_zmq_path, supports_kw, swap_dict_values,
34-
unique_filepath)
3526

27+
# isort: off
28+
from vllm.utils import (
29+
CacheInfo, FlexibleArgumentParser, LRUCache, MemorySnapshot,
30+
PlaceholderModule, bind_kv_cache, common_broadcastable_dtype,
31+
current_stream, deprecate_kwargs, get_open_port, get_tcp_uri,
32+
is_lossless_cast, join_host_port, make_zmq_path, make_zmq_socket,
33+
memory_profiling, merge_async_iterators, sha256, split_host_port,
34+
split_zmq_path, supports_kw, swap_dict_values, unique_filepath)
35+
# isort: on
3636
from ..utils import create_new_process_for_each_test, error_on_warning
3737

3838

@@ -1036,13 +1036,11 @@ def test_load_config_file(tmp_path):
10361036

10371037

10381038
def test_unique_filepath():
1039-
# make temp directory
10401039
temp_dir = tempfile.mkdtemp()
10411040
path_fn = lambda i: Path(temp_dir) / f"file_{i}.txt"
10421041
paths = set()
10431042
for i in range(10):
10441043
path = unique_filepath(path_fn)
1045-
# write to file
10461044
path.write_text("test")
10471045
paths.add(path)
10481046
assert len(paths) == 10

0 commit comments

Comments
 (0)