Merge pull request #584 from ggmarshall/main

fix for bug in timecorr
legend-exp · May 26, 2024 · 63dd3dc · 63dd3dc
2 parents 0714ada + ce01a7f
commit 63dd3dc
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/pygama/pargen/AoE_cal.py b/src/pygama/pargen/AoE_cal.py
@@ -1023,11 +1023,8 @@ def time_correction(
 
                     elif mode == "none":
                         time_dict = {
-                            time: mean
-                            for time, mean in zip(
-                                np.array(self.timecorr_df.index),
-                                np.nanmean(self.timecorr_df["mean"]),
-                            )
+                            time: np.nanmean(self.timecorr_df["mean"])
+                            for time in np.array(self.timecorr_df.index)
                         }
 
                     else: