File size: 531 Bytes
acd4009
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import jsonlines
from tqdm import tqdm
from collections import Counter, defaultdict

stat = defaultdict(Counter)

with jsonlines.open("api_aux_test_submission_java-evaluated.jsonl") as jrp:
    for sample in tqdm(jrp):
        verdict = "PASSED"
        for ut in sample["unittests"]:
            if ut["exec_outcome"] != "PASSED":
                verdict = ut["exec_outcome"]
                break
        stat[sample["lang_cluster"]][f"{sample['exec_outcome']}-{verdict}"] += 1

import json
print(json.dumps(stat, indent=4))