def sorted_node_list()

in cluster-trace-gpu-v2020/simulator/scheduler.py [0:0]

• 13 lines of code
• 5 McCabe index (conditional complexity)

    def sorted_node_list(self, node_list):
        policy = self.sort_node_policy
        if policy == 0:
            node_list.sort(key=lambda n: n.id)  # by id
        elif policy == 1:
            node_list.sort(key=lambda n: n.idl_gpus)  # smallest idle gpus first
        elif policy == 2:
            node_list.sort(key=lambda n: -n.idl_gpus)  # largest idle gpus first
        elif policy == 3:
            node_list.sort(key=lambda n: n.util_rate)  # lowest avg. util. first
        else:
            node_list.sort(key=lambda n: n.id)
        return node_list