From 358c328d69f62c1fdbd6deaf3b872809734c37b2 Mon Sep 17 00:00:00 2001 From: Zhuohan Li Date: Thu, 28 Dec 2023 06:18:11 +0800 Subject: [PATCH] [BUGFIX] Fix communication test (#2285) --- tests/distributed/test_comm_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/distributed/test_comm_ops.py b/tests/distributed/test_comm_ops.py index 733c7395..b9895b3e 100644 --- a/tests/distributed/test_comm_ops.py +++ b/tests/distributed/test_comm_ops.py @@ -8,7 +8,7 @@ import pytest import torch from vllm.config import ParallelConfig -from vllm.engine.ray_utils import get_open_port +from vllm.utils import get_open_port from vllm.model_executor.parallel_utils.communication_op import ( tensor_model_parallel_all_reduce, tensor_model_parallel_all_gather,