Bugfix in debug plot.

pes_to_spec/model.py

@@ -131,7 +131,7 @@ class Model(object):
         """
         # reduce on channel and on train ID
         sum_low_res = - np.mean(sum(list(low_res_data.values())), axis=0)
-        widths = np.arange(10, 50, step=5)
+        widths = np.arange(10, 50, step=1)
         peak_idx = find_peaks_cwt(sum_low_res, widths)
         if len(peak_idx) < 1:
             raise PromptNotFoundError()
@@ -158,7 +158,7 @@ class Model(object):
         ax.set(title="",
                xlabel="Photon Spectrometer channel",
                ylabel="Sum of all Photon Spectrometer channels")
-        plt.axvline(100,
+        plt.axvline(peak_idx,
                 linewidth=3,
                 ls="--",
                 color='r',

scripts/test_analysis.py

@@ -111,7 +111,7 @@ def main():
                  n_pca_hr=20,
                  high_res_sigma=0.2,
                  tof_start=None,
-                 delta_tof=400,
+                 delta_tof=200,
                  validation_size=0.05)
 
     train_idx = np.isin(tids, train_tids)