From ce01a7fd97409e358659d5cd69e9a96ab4235cc5 Mon Sep 17 00:00:00 2001
From: ggmarshall <george.marshall.20@ucl.ac.uk>
Date: Fri, 24 May 2024 23:58:58 +0200
Subject: [PATCH] fix timecorr none

---
 src/pygama/pargen/AoE_cal.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/src/pygama/pargen/AoE_cal.py b/src/pygama/pargen/AoE_cal.py
index edfd1e74f..8a531d2ee 100644
--- 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: