Function write_file

Megatron-LM/scripts/split_json.py:78–82

(lines, path)

        write_mapping_file(m, path)

def write_file(lines, path):
    print('Writing:', path)
    with open(path, 'w') as f:
        for l in lines:
            f.write(l+'\n')

def write_mapping_file(m, path):
    path = path+'.map'