We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
are_arrays_equal
1 parent 0acad54 commit 95c3a2cCopy full SHA for 95c3a2c
1 file changed
stumpy/core.py
@@ -214,10 +214,7 @@ def are_arrays_equal(a, b): # pragma: no cover
214
if id(a) == id(b):
215
return True
216
217
- if a.shape != b.shape:
218
- return False
219
-
220
- return ((a == b) | (np.isnan(a) & np.isnan(b))).all()
+ return np.array_equal(a, b, equal_nan=True)
221
222
223
def are_distances_too_small(a, threshold=10e-6): # pragma: no cover
0 commit comments