o `Û·i ã @s€ddlmZmZmZddlZddlmZ d deededeeeeefffdd „Z deedeeeeefffd d „Z dS) é)ÚAnyÚDictÚListN)Ú ObjectReféÿÿÿÿÚobj_refsÚ timeout_msÚreturncCs$t ¡stdƒ‚tjjjj ||¡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.)ÚrayÚis_initializedÚ RuntimeErrorÚ_privateÚworkerÚ global_workerÚ core_workerÚget_object_locations)rr©rúP/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/experimental/locations.pyrs  ÿrcCs&t ¡stdƒ‚tjjjj}| |¡S)a¿Lookup 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 rrrrÚget_local_object_locations)rrrrrr-s  r)r) Útypingrrrr Ú ray._rayletrÚintÚstrrrrrrrÚs  ÿÿÿ þ&ÿþ