Refine value to handle imprecision

TDAmeritrade · Aug 2, 2022 · 1d0a22c · 1d0a22c
1 parent e1c9785
commit 1d0a22c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stumpy/stump.py b/stumpy/stump.py
@@ -190,6 +190,9 @@ def _compute_diagonal(
                 if T_B_subseq_isconstant[i + k] and T_A_subseq_isconstant[i]:
                     pearson = 1.0
 
+                if pearson > config.STUMPY_PERFECT_CORRELATION:
+                    pearson = 1.0
+
                 if pearson > ρ[thread_idx, i, 0]:
                     ρ[thread_idx, i, 0] = pearson
                     I[thread_idx, i, 0] = i + k