diff --git a/tests/distributed/test_comm_ops.py b/tests/distributed/test_comm_ops.py index 733c7395811ef..b9895b3e71794 100644 --- a/tests/distributed/test_comm_ops.py +++ b/tests/distributed/test_comm_ops.py @@ -8,7 +8,7 @@ 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,