wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 23:11:40 +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/state.py
+2 -5
View File
@@ -703,12 +703,9 @@ class GlobalState:
heartbeat_data = pub_message.data
message = gcs_utils.HeartbeatTableData.FromString(heartbeat_data)
# Calculate available resources for this client
num_resources = len(message.resources_available_label)
dynamic_resources = {}
for i in range(num_resources):
resource_id = message.resources_available_label[i]
dynamic_resources[resource_id] = (
message.resources_available_capacity[i])
for resource_id, capacity in message.resources_available.items():
dynamic_resources[resource_id] = capacity
# Update available resources for this client
client_id = ray.utils.binary_to_hex(message.client_id)