Unverified Commit 95560d7d by ExtremeViscent Committed by GitHub

fix: Typo in the rank zero code (#240)

The original Ray controller method `execute_rank_zero_sync()` is not
functional. Fixed.
parent bf9be15c
......@@ -317,7 +317,7 @@ class RayWorkerGroup(WorkerGroup):
return new_worker_group_dict
def execute_rank_zero_sync(self, method_name: str, *args, **kwargs):
return ray.get(self.execute_all_async(method_name, **args, **kwargs))
return ray.get(self.execute_rank_zero_async(method_name, *args, **kwargs))
def execute_rank_zero_async(self, method_name: str, *args, **kwargs):
remote_call = getattr(self._workers[0], method_name)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment