Compatible with XENONnT/straxen#1513

XENONnT · Jan 15, 2025 · 5f2d076 · 5f2d076
1 parent 0adbc53
commit 5f2d076
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/axidence/plugins/salting/peak_correlation.py b/axidence/plugins/salting/peak_correlation.py
@@ -28,9 +28,10 @@ def infer_dtype(self):
         return dtype
 
     def compute(self, peaks_salted, peaks):
-        windows = strax.touching_windows(peaks, peaks_salted, window=self.nearby_window)
+        mask = np.isin(peaks["type"], [1, 2])
+        windows = strax.touching_windows(peaks[mask], peaks_salted, window=self.nearby_window)
         n_left, n_tot = self.find_n_competing(
-            peaks, peaks_salted, windows, fraction=self.min_area_fraction
+            peaks[mask], peaks_salted, windows, fraction=self.min_area_fraction
         )
         return dict(
             time=peaks_salted["time"],