diff --git a/PeakOTron.py b/PeakOTron.py index 9d315483340802ac8146b3181ee2e6eb359d3104..56fa80019168634def97f981991d000d00af4f41 100644 --- a/PeakOTron.py +++ b/PeakOTron.py @@ -410,9 +410,7 @@ class PeakOTron: y_hat_all = self.GetModel(x_all, prefit=prefit, bin_units=True) else: y_hat_all = DRM(x_all, **pars) - - y_hat_err_all = np.sqrt(y_hat_all) - + conds = [] cond_count = (y_all>1) @@ -442,7 +440,7 @@ class PeakOTron: y = y_all[conds] y_err = y_err_all[conds] y_hat = y_hat_all[conds] - y_hat_err = y_hat_err_all[conds] + y_hat_err = np.sqrt(y_hat_all[conds])