From c5b6dc997a933d24866f3bdba659cc26a5e7278d Mon Sep 17 00:00:00 2001 From: dachengx Date: Wed, 19 Mar 2025 00:37:19 -0500 Subject: [PATCH] Be compatible to https://github.com/AxFoundation/strax/pull/990 --- straxen/plugins/peaks/peak_ambience.py | 2 +- straxen/plugins/peaks/peak_nearest_triggering.py | 2 +- straxen/plugins/peaks/peak_se_score.py | 2 +- straxen/plugins/peaks/peak_shadow.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/straxen/plugins/peaks/peak_ambience.py b/straxen/plugins/peaks/peak_ambience.py index a5f9a78e0..0871058c9 100644 --- a/straxen/plugins/peaks/peak_ambience.py +++ b/straxen/plugins/peaks/peak_ambience.py @@ -76,7 +76,7 @@ def infer_dtype(self): return dtype def compute(self, lone_hits, peaks): - argsort = strax.stable_argsort(peaks["center_time"]) + argsort = strax.stable_argsort(peaks, order="center_time") _peaks = strax.stable_sort(peaks, order="center_time") result = np.zeros(len(peaks), self.dtype) _quick_assign(argsort, result, self.compute_ambience(lone_hits, peaks, _peaks)) diff --git a/straxen/plugins/peaks/peak_nearest_triggering.py b/straxen/plugins/peaks/peak_nearest_triggering.py index 44435d47c..18bd30a37 100644 --- a/straxen/plugins/peaks/peak_nearest_triggering.py +++ b/straxen/plugins/peaks/peak_nearest_triggering.py @@ -61,7 +61,7 @@ def get_window_size(self): return 10 * self.shadow_time_window_backward def compute(self, peaks): - argsort = strax.stable_argsort(peaks["center_time"]) + argsort = strax.stable_argsort(peaks, order="center_time") _peaks = strax.stable_sort(peaks, order="center_time") result = np.zeros(len(peaks), self.dtype) _quick_assign(argsort, result, self.compute_triggering(peaks, _peaks)) diff --git a/straxen/plugins/peaks/peak_se_score.py b/straxen/plugins/peaks/peak_se_score.py index b70078d10..8074d351f 100644 --- a/straxen/plugins/peaks/peak_se_score.py +++ b/straxen/plugins/peaks/peak_se_score.py @@ -147,7 +147,7 @@ def compute_se_score(self, peaks, _peaks): def compute(self, peaks): # sort peaks by center_time - argsort = strax.stable_argsort(peaks["center_time"]) + argsort = strax.stable_argsort(peaks, order="center_time") _peaks = strax.stable_sort(peaks, order="center_time") mask_nan = np.isnan(_peaks["x"]) | np.isnan(_peaks["y"]) # prepare output diff --git a/straxen/plugins/peaks/peak_shadow.py b/straxen/plugins/peaks/peak_shadow.py index 20988bbbd..50b93a1ee 100644 --- a/straxen/plugins/peaks/peak_shadow.py +++ b/straxen/plugins/peaks/peak_shadow.py @@ -145,7 +145,7 @@ def shadowdtype(self): return dtype def compute(self, peaks): - argsort = strax.stable_argsort(peaks["center_time"]) + argsort = strax.stable_argsort(peaks, order="center_time") _peaks = strax.stable_sort(peaks, order="center_time") result = np.zeros(len(peaks), self.dtype) _quick_assign(argsort, result, self.compute_shadow(peaks, _peaks))