File size: 397 Bytes
5920b49
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import os
import json
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--src", type=str)
parser.add_argument("--dst", type=str)
args = parser.parse_args()

cur_result = {}

for line in open(args.src):
    data = json.loads(line)
    qid = data['question_id']
    cur_result[f'v1_{qid}'] = data['text']

with open(args.dst, 'w') as f:
    json.dump(cur_result, f, indent=2)