add no timecorr mode, bugfix for dt corr plot

legend-exp · May 24, 2024 · c3afc87 · c3afc87
1 parent b162031
commit c3afc87
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/pygama/pargen/AoE_cal.py b/src/pygama/pargen/AoE_cal.py
@@ -957,6 +957,15 @@ 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"]),
+                            )
+                        }
+
                     else:
                         raise ValueError("unknown mode")
 
@@ -2005,7 +2014,7 @@ def drifttime_corr_plot(
 
         hist, bins, var = pgh.get_hist(
             final_df[aoe_class.dt_param],
-            dx=10,
+            dx=32,
             range=(
                 np.nanmin(final_df[aoe_class.dt_param]),
                 np.nanmax(final_df[aoe_class.dt_param]),