Fixed #501 Allow max_distance = np.inf in match

seanlaw · seanlaw · commit afc79f98852a · 2021-12-21T23:00:35.000-05:00
diff --git a/stumpy/aamp_motifs.py b/stumpy/aamp_motifs.py
@@ -373,7 +373,11 @@ def max_distance(D):
     matches = []
 
     candidate_idx = np.argmin(D)
-    while D[candidate_idx] <= atol + max_distance and len(matches) < max_matches:
+    while (
+        D[candidate_idx] <= atol + max_distance
+        and np.isfinite(D[candidate_idx])
+        and len(matches) < max_matches
+    ):
         matches.append([D[candidate_idx], candidate_idx])
         core.apply_exclusion_zone(D, candidate_idx, excl_zone)
         candidate_idx = np.argmin(D)
diff --git a/stumpy/motifs.py b/stumpy/motifs.py
@@ -414,7 +414,11 @@ def max_distance(D):
     matches = []
 
     candidate_idx = np.argmin(D)
-    while D[candidate_idx] <= atol + max_distance and len(matches) < max_matches:
+    while (
+        D[candidate_idx] <= atol + max_distance
+        and np.isfinite(D[candidate_idx])
+        and len(matches) < max_matches
+    ):
         matches.append([D[candidate_idx], candidate_idx])
         core.apply_exclusion_zone(D, candidate_idx, excl_zone)
         candidate_idx = np.argmin(D)
