Minor test

seanlaw · seanlaw · commit 724cf8535a85 · 2022-03-03T18:31:39.000-05:00
diff --git a/stumpy/scrump.py b/stumpy/scrump.py
@@ -95,19 +95,19 @@ def _compute_PI(
     """
     l = T_B.shape[0] - m + 1
     squared_distance_profile = np.empty(l)
+    QT = np.empty(l, dtype=np.float64)
     for i in indices[start:stop]:
         Q = T_A[i : i + m]
-        # QT = core._sliding_dot_product(Q, T_B)
-        QT = np.empty(l, dtype=np.float64)
+        # QT[:] = core._sliding_dot_product(Q, T_B)
         for j in range(l):
             QT[j] = np.dot(Q, T_B[j : j + m])
         # Update P[i] relative to all T[j : j + m]
         squared_distance_profile[:] = core._mass(Q, T_B, QT, μ_Q[i], σ_Q[i], M_T, Σ_T)
-        squared_distance_profile[:] = np.square(squared_distance_profile)
-        if excl_zone is not None:
-            zone_start = max(0, i - excl_zone)
-            zone_stop = min(l, i + excl_zone)
-            squared_distance_profile[zone_start : zone_stop + 1] = np.inf
+        # squared_distance_profile[:] = np.square(squared_distance_profile)
+        # if excl_zone is not None:
+        #     zone_start = max(0, i - excl_zone)
+        #     zone_stop = min(l, i + excl_zone)
+        #     squared_distance_profile[zone_start : zone_stop + 1] = np.inf
         # I[thread_idx, i] = np.argmin(squared_distance_profile)
         # P_squared[thread_idx, i] = squared_distance_profile[I[thread_idx, i]]
         # if P_squared[thread_idx, i] == np.inf:  # pragma: no cover
