Add code to resolve warning

Author: NimaSarajpoor

stumpy/core.py

@@ -2254,6 +2254,7 @@ def preprocess_diagonal(
 
     M_T, Σ_T = compute_mean_std(T, m)
     Σ_T[T_subseq_isconstant] = 1.0  # Avoid divide by zero in next inversion step
+    Σ_T[~T_subseq_isfinite] = 1.0
     Σ_T_inverse = 1.0 / Σ_T
     M_T_m_1, _ = compute_mean_std(T, m - 1)
 

tests/test_core.py

@@ -881,9 +881,11 @@ def test_preprocess_non_normalized():
 def test_preprocess_diagonal():
     T = np.array([0, np.nan, 2, 3, 4, 5, 6, 7, np.inf, 9])
     m = 3
+    T_subseq_isfinite = core.rolling_isfinite(T, m)
 
     ref_T = np.array([0, 0, 2, 3, 4, 5, 6, 7, 0, 9], dtype=float)
     ref_M, ref_Σ = naive.compute_mean_std(ref_T, m)
+    ref_Σ[~T_subseq_isfinite] = 1.0
     ref_Σ_inverse = 1.0 / ref_Σ
     ref_M_m_1, _ = naive.compute_mean_std(ref_T, m - 1)