| 12345678910111213141516171819202122232425262728 |
- 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
|