Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-11-16 06:12:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix choose_best_blocks()

Browse Source
This commit is contained in:
Aleksandr Borzunov 2022-07-13 22:59:01 +00:00
parent 75856e4769
commit f3cf5f4d8d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/server/block_selection.py
View File

@ -12,7 +12,7 @@ def choose_best_blocks(num_blocks: int, remote_module_infos: List[Optional[Remot
throughputs.append(sum(server.throughput for server in module.servers.values()
if server.state != ServerState.OFFLINE))
options = [(throughputs[i:i + num_blocks], i)
options = [(sorted(throughputs[i:i + num_blocks]), i)
for i in range(0, len(throughputs) - num_blocks + 1)]
best_start = min(options)[1]
return list(range(best_start, best_start + num_blocks))