VARestorer / infinity /utils /csv_util.py
YixuanEvan's picture
add HF model card and mirror runnable codebase
7f7272e
raw
history blame contribute delete
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)