def add_user_round_robin_id()

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

• 11 lines of code
• 1 McCabe index (conditional complexity)

def add_user_round_robin_id(job_list):
    # Add a new sorting metrics, user_rrid, to enforce scheduler picking jobs from multiple users
    # when all users' primary metrics are the same (e.g., 0).
    user_rrid_dict = {}  # a new dict each time
    for job in job_list:
        user = job['user']
        rrid = user_rrid_dict.get(user, None)
        if rrid is None:
            rrid = 0
            user_rrid_dict[user] = 1
        else:
            user_rrid_dict[user] += 1
        job['user_rrid'] = rrid