Add experimental API for ray.get and ray.wait with additional argument types (#2071)

This commit is contained in:
Kunal Gosar
2018-06-01 16:42:27 -07:00
committed by Robert Nishihara
parent 4dd4698564
commit 317d0da7d8
3 changed files with 116 additions and 1 deletions
+3 -1
View File
@@ -8,10 +8,12 @@ from .features import (
flush_finished_tasks_unsafe, flush_evicted_objects_unsafe,
_flush_finished_tasks_unsafe_shard, _flush_evicted_objects_unsafe_shard)
from .named_actors import get_actor, register_actor
from .api import get, wait
__all__ = [
"TensorFlowVariables", "flush_redis_unsafe",
"flush_task_and_object_metadata_unsafe", "flush_finished_tasks_unsafe",
"flush_evicted_objects_unsafe", "_flush_finished_tasks_unsafe_shard",
"_flush_evicted_objects_unsafe_shard", "get_actor", "register_actor"
"_flush_evicted_objects_unsafe_shard", "get_actor", "register_actor",
"get", "wait"
]