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

Update task_table and object_table API. (#3161)

Browse Source 
* Update task_table and object_table API.

* Fix
This commit is contained in:
Robert Nishihara
2018-10-31 12:52:50 -07:00
committed by Philipp Moritz
parent 9df2e6e6f4
commit 1f29a960f4
3 changed files with 54 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/monitor.py
+3 -6
View File
@@ -152,10 +152,8 @@ class Monitor(object):
task_table_objects = self.state.task_table()
driver_id_hex = binary_to_hex(driver_id)
driver_task_id_bins = set()
for task_id_hex in task_table_objects:
if len(task_table_objects[task_id_hex]) == 0:
continue
task_table_object = task_table_objects[task_id_hex][0]["TaskSpec"]
for task_id_hex, task_info in task_table_objects.items():
task_table_object = task_info["TaskSpec"]
task_driver_id_hex = task_table_object["DriverID"]
if driver_id_hex != task_driver_id_hex:
# Ignore tasks that aren't from this driver.
@@ -165,8 +163,7 @@ class Monitor(object):
# Get objects associated with the driver.
object_table_objects = self.state.object_table()
driver_object_id_bins = set()
for object_id, object_table_object in object_table_objects.items():
assert len(object_table_object) > 0
for object_id, _ in object_table_objects.items():
task_id_bin = ray.raylet.compute_task_id(object_id).id()
if task_id_bin in driver_task_id_bins:
driver_object_id_bins.add(object_id.id())