We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0d17b90 commit 53e83f1Copy full SHA for 53e83f1
rlinf/scheduler/cluster/node.py
@@ -428,7 +428,7 @@ def _sort_nodes(self, cluster_num_nodes: int):
428
# NODE_RANK not set, sort first by accelerator type, then by IP
429
nodes_group_by_accel: dict[str, list[NodeInfo]] = {}
430
for node in self._nodes:
431
- accel_name = f"{node.accelerator_type.value}_{node.accelerator_model}"
+ accel_name = node.accelerator_type
432
nodes_group_by_accel.setdefault(accel_name, [])
433
nodes_group_by_accel[accel_name].append(node)
434
for accel_name in nodes_group_by_accel.keys():
0 commit comments