wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 21:26:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Use map instead of list to represent resources in heartbeat message (#9294)

Browse Source
This commit is contained in:
Tao Wang
2020-07-05 10:59:25 +08:00
committed by GitHub
parent 28d18ec836
commit f7ac495a68
5 changed files with 34 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/monitor.py
+3 -9
View File
@@ -97,15 +97,9 @@ class Monitor:
message = ray.gcs_utils.HeartbeatBatchTableData.FromString(
heartbeat_data)
for heartbeat_message in message.batch:
resource_load = dict(
zip(heartbeat_message.resource_load_label,
heartbeat_message.resource_load_capacity))
total_resources = dict(
zip(heartbeat_message.resources_total_label,
heartbeat_message.resources_total_capacity))
available_resources = dict(
zip(heartbeat_message.resources_available_label,
heartbeat_message.resources_available_capacity))
resource_load = dict(heartbeat_message.resource_load)
total_resources = dict(heartbeat_message.resources_total)
available_resources = dict(heartbeat_message.resources_available)
for resource in total_resources:
available_resources.setdefault(resource, 0.0)