o $i @sddlmZmZmZddlZddlmZ d deededeeeeefffdd Z deedeeeeefffd d Z dS) )AnyDictListN) ObjectRefobj_refs timeout_msreturncCs$tstdtjjjj||S)adLookup the locations for a list of objects. It returns a dict maps from an object to its location. The dict excludes those objects whose location lookup failed. Args: object_refs (List[ObjectRef]): List of object refs. timeout_ms: The maximum amount of time in micro seconds to wait before returning. Wait infinitely if it's negative. Returns: A dict maps from an object to its location. The dict excludes those objects whose location lookup failed. The location is stored as a dict with following attributes: - node_ids (List[str]): The hex IDs of the nodes that have a copy of this object. Objects less than 100KB will be in memory store not plasma store and therefore will have nodes_id = []. - object_size (int): The size of data + metadata in bytes. Can be None if the size is unknown yet (e.g. task not completed). Raises: RuntimeError: if the processes were not started by ray.init(). ray.exceptions.GetTimeoutError: if it couldn't finish the request in time. Ray hasn't been initialized.)rayis_initialized RuntimeError_privateworker global_worker core_workerget_object_locations)rrrW/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/experimental/locations.pyrs  rcCs&tstdtjjjj}||S)aLookup the locations for a list of objects *from the local core worker*. No RPCs are made in this method. It returns a dict maps from an object to its location. The dict excludes those objects whose location lookup failed. Args: object_refs (List[ObjectRef]): List of object refs. Returns: A dict maps from an object to its location. The dict excludes those objects whose location lookup failed. The location is stored as a dict with following attributes: - node_ids (List[str]): The hex IDs of the nodes that have a copy of this object. Objects less than 100KB will be in memory store not plasma store and therefore will have nodes_id = []. - object_size (int): The size of data + metadata in bytes. Can be None if the size is unknown yet (e.g. task not completed). Raises: RuntimeError: if the processes were not started by ray.init(). r )r r r rrrrget_local_object_locations)rrrrrr-s  r)r) typingrrrr ray._rayletrintstrrrrrrrs   &