[Tests] Use get_file_store_init_method helper from vllm (#7528)

Signed-off-by: Nick Cao <ncao@redhat.com>
Co-authored-by: Codex <noreply@openai.com>
This commit is contained in:
Nick CaoandCodex authored and GitHub committed 2026-09-28 14:22:25 -04:00
1 parent 12e9280010
commit b656fcf45b
8 files changed
+28 -42

No files matched your search

+3 -3
View File
@@ -9,8 +9,8 @@ from types import SimpleNamespace
import pytest
import torch
from vllm.utils.network_utils import get_file_store_init_method
from tests.helpers.runtime import get_distributed_init_method
from vllm_omni.diffusion.cache.seacache import (
SeaCacheBackend,
SeaCacheConfig,
@@ -375,7 +375,7 @@ def test_parameter_sharded_hook_handles_uneven_cfg_branch_dispatch() -> None:
result_queue = manager.Queue()
torch.multiprocessing.spawn(
_uneven_cfg_sharded_worker,
args=(get_distributed_init_method("seacache_uneven_cfg_"), result_queue),
args=(get_file_store_init_method(), result_queue),
nprocs=4,
)
@@ -452,7 +452,7 @@ def test_hook_synchronizes_full_hybrid_sequence_parallel_group() -> None:
result_queue = manager.Queue()
torch.multiprocessing.spawn(
_hybrid_sp_worker,
args=(get_distributed_init_method("seacache_hybrid_sp_"), result_queue),
args=(get_file_store_init_method(), result_queue),
nprocs=4,
)