wehub-resource-sync
29 lines
689 B
Python
29 lines
689 B
Python
from typing import List
|
|
|
|
import ray
|
|
|
|
|
|
class CommunicatorHandle:
|
|
"""
|
|
A lightweight communicator handle used by the driver to store handles to
|
|
the actors in the communicator.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
actor_handles: List["ray.actor.ActorHandle"],
|
|
):
|
|
"""
|
|
Initializes the CommunicatorHandle with the given actor handles.
|
|
|
|
Args:
|
|
actor_handles: A list of actor handles to be stored.
|
|
"""
|
|
self._actor_handles = actor_handles
|
|
|
|
def get_actor_handles(self) -> List["ray.actor.ActorHandle"]:
|
|
"""
|
|
Retuan all actor handles in this communicator.
|
|
"""
|
|
return self._actor_handles
|