Skip to content

Commit 95c3a2c

Browse files
committed
Fixed #317 Replace comparison in are_arrays_equal
1 parent 0acad54 commit 95c3a2c

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

stumpy/core.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -214,10 +214,7 @@ def are_arrays_equal(a, b): # pragma: no cover
214214
if id(a) == id(b):
215215
return True
216216

217-
if a.shape != b.shape:
218-
return False
219-
220-
return ((a == b) | (np.isnan(a) & np.isnan(b))).all()
217+
return np.array_equal(a, b, equal_nan=True)
221218

222219

223220
def are_distances_too_small(a, threshold=10e-6): # pragma: no cover

0 commit comments

Comments
 (0)