[Fix] fix import error of RayWorker (#604) (#605)

2023-07-28 14:37:40 +08:00
parent 58a072be15
commit c0d00f5be6
1 changed files with 1 additions and 0 deletions
--- a/vllm/engine/ray_utils.py
+++ b/vllm/engine/ray_utils.py
@@ -27,6 +27,7 @@ try:
 except ImportError:
    ray = None
    TorchDistributedWorker = None
+    RayWorker = None  # pylint: disable=invalid-name

 if TYPE_CHECKING:
    from ray.util.placement_group import PlacementGroup