dream

1
import argparse
2
import json
3

4
parser = argparse.ArgumentParser()
5
parser.add_argument("--json_infile", type=str, default="test_question_tasks_results.json", help="from tasks json file")
6
parser.add_argument("--json_outfile", type=str, default="toloka_test_question_tasks_results.json", help="output file")
7

8

9
def main():
10
    args = parser.parse_args()
11
    cntx = json.load(open(args.json_infile, "rt"))
12
    tasks = []
13
    for task in cntx["tasks"]:
14
        responses = {
15
            resp.get("response", ""): resp
16
            for resp in task["responses"]
17
            if resp.get("response") and resp.get("response") != "sorry"
18
        }
19
        responses = sorted(responses.values(), key=lambda x: -x.get("confidence", 0))
20
        task["responses"] = responses
21
        tasks.append(task)
22

23
    json.dump(cntx, open(args.json_outfile, "wt", encoding="utf-8"), indent=4)
24

25

26
if __name__ == "__main__":
27
    main()
28