mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 02:01:24 +08:00
Sven
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
import ray
|
|
import numpy as np
|
|
|
|
|
|
def get(object_ids):
|
|
"""Get a single or a collection of remote objects from the object store.
|
|
|
|
This method is identical to `ray.get` except it adds support for tuples,
|
|
ndarrays and dictionaries.
|
|
|
|
Args:
|
|
object_ids: Object ID of the object to get, a list, tuple, ndarray of
|
|
object IDs to get or a dict of {key: object ID}.
|
|
|
|
Returns:
|
|
A Python object, a list of Python objects or a dict of {key: object}.
|
|
"""
|
|
if isinstance(object_ids, (tuple, np.ndarray)):
|
|
return ray.get(list(object_ids))
|
|
elif isinstance(object_ids, dict):
|
|
keys_to_get = [
|
|
k for k, v in object_ids.items() if isinstance(v, ray.ObjectID)
|
|
]
|
|
ids_to_get = [
|
|
v for k, v in object_ids.items() if isinstance(v, ray.ObjectID)
|
|
]
|
|
values = ray.get(ids_to_get)
|
|
|
|
result = object_ids.copy()
|
|
for key, value in zip(keys_to_get, values):
|
|
result[key] = value
|
|
return result
|
|
else:
|
|
return ray.get(object_ids)
|
|
|
|
|
|
def wait(object_ids, num_returns=1, timeout=None):
|
|
"""Return a list of IDs that are ready and a list of IDs that are not.
|
|
|
|
This method is identical to `ray.wait` except it adds support for tuples
|
|
and ndarrays.
|
|
|
|
Args:
|
|
object_ids (List[ObjectID], Tuple(ObjectID), np.array(ObjectID)):
|
|
List like of object IDs for objects that may or may not be ready.
|
|
Note that these IDs must be unique.
|
|
num_returns (int): The number of object IDs that should be returned.
|
|
timeout (float): The maximum amount of time in seconds to wait before
|
|
returning.
|
|
|
|
Returns:
|
|
A list of object IDs that are ready and a list of the remaining object
|
|
IDs.
|
|
"""
|
|
if isinstance(object_ids, (tuple, np.ndarray)):
|
|
return ray.wait(
|
|
list(object_ids), num_returns=num_returns, timeout=timeout)
|
|
|
|
return ray.wait(object_ids, num_returns=num_returns, timeout=timeout)
|