o $i @sddlmZmZmZddlZdedfddZdddeedefdeeedefeedefffd d Zd ed eedefdeeedefeedefffd dZdeddefddZ dS))ListOptionalTupleNreturnzray.actor.ActorHandlecCs"ztjWStyYdSw)zv Get the current actor handle in this worker. If this is called in a driver process, it will return None. N)rayget_runtime_context current_actor RuntimeErrorr r [/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/experimental/channel/utils.pyget_self_actors   r writerreader_and_node_listcCsDg}g}|D]\}}||kr|||fq|||fq||fS)a\Split readers into remote and local readers based on writer. Args: writer: The actor handle of the writer reader_and_node_list: List of (reader, node) tuples Returns: Tuple containing: - List of (reader, node) tuples for remote readers - List of (reader, node) tuples for local readers append)r rremote_readers local_readersreadernoder r r split_readers_by_localitys rractor_and_node_listcCsDg}g}|D]\}}||kr|||fq|||fq||fS)aSplit actors into remote and local actors based on node. The local actors will be on the same node as the given node. The remote actors will be on a different node. Args: writer_node: The node of the writer actor_and_node_list: List of (actor, node) tuples Returns: Tuple containing: - List of (actor, node) tuples for actors on the same node - List of (actor, node) tuples for actors on a different node r)rractors_on_same_nodeactors_on_different_nodeactor actor_noder r r split_actors_by_node_locality.s rrcCs8|dus |tjkrtSt|jddS)zGet the node of the actor. Args: actor: The actor handle of the actor Returns: The node of the actor NcSs tS)N)rr get_node_id)selfr r r Zs z get_actor_node..)rrrrget __ray_call__remote)rr r r get_actor_nodeLs r") typingrrrrr strrrr"r r r r s,