From 8c617ee2f2ba620debeb3b5c3d42f490206c4051 Mon Sep 17 00:00:00 2001 From: Erty Seidohl Date: Thu, 9 Feb 2023 15:09:11 -0500 Subject: [PATCH] Respect delimiter choice in get_loadings Previously hardcoded to '\t' which caused errors when trying to load csv files that used other delimiters. --- ddr/get_loadings.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ddr/get_loadings.py b/ddr/get_loadings.py index a41b0f9..a82eb2f 100644 --- a/ddr/get_loadings.py +++ b/ddr/get_loadings.py @@ -38,10 +38,10 @@ def get_loadings(agg_doc_vecs_path, agg_dic_vecs_path, out_path, num_features, d with open(agg_doc_vecs_path, 'rb') as doc_vecs, open(out_path, 'wb') as out_file: - doc_vecs_reader = csv.reader(doc_vecs, delimiter='\t') + doc_vecs_reader = csv.reader(doc_vecs, delimiter=delimiter) doc_vecs_reader.next() - writer = csv.writer(out_file, delimiter='\t') + writer = csv.writer(out_file, delimiter=delimiter) fieldnames_out = ['ID'] + list(dic_vecs.keys()) writer.writerow(fieldnames_out)