Spaces:
Build error
Build error
import json | |
from argparse import ArgumentParser | |
if __name__ == '__main__': | |
parser = ArgumentParser() | |
parser.add_argument( | |
'log_fname', | |
type=str, | |
) | |
args = parser.parse_args() | |
fname = args.log_fname | |
out_fname = args.log_fname.replace('.jsonl', '.converted.jsonl') | |
log = [json.loads(line) for line in open(fname)] | |
simple_log = [ | |
json.dumps( | |
{ | |
'instance_id': ex['instance_id'], | |
'instruction': ex['instruction'], | |
'test_result': ex['test_result'], | |
'cost': ex['metrics']['accumulated_cost'], | |
} | |
) | |
for ex in log | |
] | |
with open(out_fname, 'w+', encoding='utf-8') as f: | |
f.write('\n'.join(simple_log)) | |