chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,28 @@
|
||||
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
|
||||
Reference in New Issue
Block a user