Skip to content

Commit e0fb52d

Browse files
authored
fix: failing test I missed (#4162)
Signed-off-by: alec-flowers <[email protected]>
1 parent 209783d commit e0fb52d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/fault_tolerance/etcd_ha/test_vllm.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,10 @@ def __init__(self, request, etcd_endpoints: list, is_prefill: bool = False):
6464
env["DYN_SYSTEM_USE_ENDPOINT_HEALTH_STATUS"] = '["generate"]'
6565
env["DYN_SYSTEM_PORT"] = port
6666

67+
if is_prefill:
68+
env["DYN_VLLM_KV_EVENT_PORT"] = "20082"
69+
env["VLLM_NIXL_SIDE_CHANNEL_PORT"] = "5601"
70+
6771
# Set log directory based on worker type
6872
worker_type = "prefill_worker" if is_prefill else "worker"
6973
log_dir = f"{request.node.name}_{worker_type}"

0 commit comments

Comments
 (0)