File size: 569 Bytes
50b41da | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import jsonlines
results = {}
with jsonlines.open("/home/weifengsun/tangou1/step2/step22/output/gemini_results.jsonl", "r") as reader:
for obj in reader:
results[obj["idx"]] = obj
with jsonlines.open("/home/weifengsun/tangou1/step2/step22/output/function_filtered_scores.jsonl", "r") as reader, jsonlines.open("/home/weifengsun/tangou1/step2/step22/output/alignment.jsonl", "w") as writer:
for idx, obj in enumerate(reader):
if idx not in results:
continue
obj["results"] = results[idx]
writer.write(obj)
|