def _key_by_problem()

in prm800k/eval/eval.py [0:0]

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

def _key_by_problem(samples: List[Dict]):
    grouped_samples = defaultdict(list)
    for sample in samples:
        grouped_samples[sample["problem"]].append(sample)
    return grouped_samples