o $ik@sPddlmZddlmZddlZddlmZeGdddZGdddZdS) ) dataclass)ListN)Backendc@s*eZdZUdZeed<eed<eed<dS) Communicatorz= A handle to a communicator that we are a member of. namerankbackendN)__name__ __module__ __qualname____doc__str__annotations__intrre/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/experimental/collective/communicator.pyrs  rc@s~eZdZdZdeejjdedefddZ dejjfdd Z e d eejjfd d Z e d efd dZ e d efddZdS)CommunicatorHandlezj A communicator handle used by the driver to store handles to the actors in the communicator. actorsrrcCs||_||_t||_dS)a Initializes the CommunicatorHandle with the given actor handles. Assumes that the communicator has already been initialized on all actors. Args: actors: A list of actor handles to be stored. name: Name of the communicator. backend: Communicator backend. See ray.util.collective.types for valid values. N)_actors_namer_backend)selfrrrrrr__init__s zCommunicatorHandle.__init__actorcCs(t|jD] \}}||kr|SqdS)N) enumerater)rriarrrget_rank,s zCommunicatorHandle.get_rankreturncCs|jddS)z@ Return all actor handles in this communicator. N)rrrrrr2szCommunicatorHandle.actorscC|jSN)rr rrrr9zCommunicatorHandle.namecCr!r")rr rrrr=r#zCommunicatorHandle.backendN)r r r r rrayr ActorHandler rrpropertyrrrrrrrrsr) dataclassesrtypingrr$ray.util.collective.typesrrrrrrrs