| 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) | |