Support Gram with trailing delim

barakcohenlab · Mar 17, 2020 · fb11951 · fb11951
1 parent f86a567
commit fb11951
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/preimage/kernels/generic_string.py b/preimage/kernels/generic_string.py
@@ -369,7 +369,7 @@ def set_gram_matrix_from_file(self, filename, delim="\t", lower_triangle=True):
             gram_matrix = np.eye(n_lines)
             with open(filename) as fin:
                 for row, line in enumerate(fin):
-                    line = map(float, line.split(delim))
+                    line = map(float, line.strip().split(delim))
                     for col, val in enumerate(line):
                         if row != col:
                             gram_matrix[row, col] = val