Compatible with XENONnT/straxen#1513 (#98)

* Compatible with XENONnT/straxen#1513 * Inherit `find_n_competing`
XENONnT · Feb 7, 2025 · 6131a94 · 6131a94
1 parent e73d316
commit 6131a94
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
@@ -27,9 +27,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"],