[BUGFIX] Fix communication test (vllm-project#2285)

esmeetu · Dec 27, 2023 · 358c328 · 358c328
1 parent 4aaafdd
commit 358c328
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,