From 28c4d09887ed7a8f99edf0a8e6b74f338e85f4c2 Mon Sep 17 00:00:00 2001 From: Yair Schiff Date: Tue, 8 Sep 2020 09:54:53 -0400 Subject: [PATCH] Fix flag check in add_col method --- pyrivet/hilbert_distance.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyrivet/hilbert_distance.py b/pyrivet/hilbert_distance.py index 88b6198..9e8841b 100644 --- a/pyrivet/hilbert_distance.py +++ b/pyrivet/hilbert_distance.py @@ -117,7 +117,7 @@ def add_col(self, first_length): mat = np.insert(self.mat, 0, [0], axis=1) # print("After:") # print(mat) - elif col == DimensionQueryResult.HIGH: + elif flag == DimensionQueryResult.HIGH: # print("HIGH") mat = np.append(self.mat, np.zeros((self.mat.shape[0], 1)), axis=1) elif not self.dimensions[1].is_bound(first_length): @@ -235,4 +235,4 @@ def betti_to_splitmat(betti: rivet.MultiBetti): def distance(left: rivet.MultiBetti, right: rivet.MultiBetti): - return betti_to_splitmat(left).distance(betti_to_splitmat(right)) \ No newline at end of file + return betti_to_splitmat(left).distance(betti_to_splitmat(right))