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

This commit is contained in:
Xudong Zhang
2023-07-27 23:37:40 -07:00
committed by GitHub
parent 58a072be15
commit c0d00f5be6
+1
View File
@@ -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