diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..0a1b85c --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/csv/history.tsv diff --git a/csv/history.tsv b/csv/history.tsv deleted file mode 100644 index 414d607..0000000 --- a/csv/history.tsv +++ /dev/null @@ -1 +0,0 @@ -model_A model_A_hash model_B model_B_hash model_O model_O_hash base_alpha weight_name weight_values \ No newline at end of file diff --git a/scripts/merge_history.py b/scripts/merge_history.py index e352057..1aa3d11 100644 --- a/scripts/merge_history.py +++ b/scripts/merge_history.py @@ -32,8 +32,8 @@ class MergeHistory(): if not os.path.exists(self.filepath): with open(self.filepath, "w", newline="", encoding="utf-8") as f: - wr = writer(f, fieldnames=HEADERS, delimiter='\t') - wr.writerow(HEADERS) + wr = DictWriter(f, fieldnames=HEADERS, delimiter='\t') + wr.writeheader() # save to file with open(self.filepath, "a", newline="", encoding='utf-8') as f: dictwriter = DictWriter(f, fieldnames=HEADERS, delimiter='\t')