wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-02 17:23:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a nicer warning message when you pass the wrong thing to ray.wait() (#1239)

Browse Source 
* add warnings

* fix python mode

* Small changes and add tests.

* Fix test failure.
This commit is contained in:
Eric Liang
2017-11-27 22:57:33 -08:00
committed by Robert Nishihara
parent c1496b8111
commit 37831ae0c3
3 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/worker.py
+16
View File
@@ -2223,6 +2223,22 @@ def wait(object_ids, num_returns=1, timeout=None, worker=global_worker):
A list of object IDs that are ready and a list of the remaining object
IDs.
"""
if isinstance(object_ids, ray.local_scheduler.ObjectID):
raise TypeError(
"wait() expected a list of ObjectID, got a single ObjectID")
if not isinstance(object_ids, list):
raise TypeError("wait() expected a list of ObjectID, got {}".format(
type(object_ids)))
if worker.mode != PYTHON_MODE:
for object_id in object_ids:
if not isinstance(object_id, ray.local_scheduler.ObjectID):
raise TypeError(
"wait() expected a list of ObjectID, "
"got list containing {}".format(type(object_id)))
check_connected(worker)
with log_span("ray:wait", worker=worker):
check_main_thread()