def preempt_job_sort_node()

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

• 7 lines of code
• 4 McCabe index (conditional complexity)

    def preempt_job_sort_node(self, node, preempt_policy):
        if preempt_policy == 1: # small_size_first
            node.job_runn_list.sort(key=lambda e: (e['size'], e['job_id']))
        elif preempt_policy == 2: # large_gang_first
            node.job_runn_list.sort(key=lambda e: (-e['num_gpu'], e['job_id']))
        else: # preempt_policy==0 or others: short_duration_first
            node.job_runn_list.sort(key=lambda e: (e['duration'], e['job_id']))