VARestorer / infinity /utils /csv_util.py
YixuanEvan's picture
add HF model card and mirror runnable codebase
7f7272e
raw
history blame
672 Bytes
import os
import os.path as osp
import csv
import numpy as np
def write_dicts2csv_file(input_dict_list, csv_filename):
os.makedirs(osp.dirname(csv_filename), exist_ok=True)
with open(csv_filename, mode='w', newline='', encoding='utf-8') as file:
fieldnames = input_dict_list[0].keys()
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader()
writer.writerows(input_dict_list)
print(f'"{csv_filename}" has been written.')
def load_csv_as_dicts(csv_filename):
with open(csv_filename, mode='r', newline='', encoding='utf-8') as csvfile:
reader = csv.DictReader(csvfile)
return list(reader)