From 2c9b17520c1c06d0fad89cc900106e5c814ddd3e Mon Sep 17 00:00:00 2001
From: Jack Christopher Hutchinson Rolph <jack.rolph@desy.de>
Date: Wed, 11 Jan 2023 12:16:36 +0100
Subject: [PATCH] Update PeakOTron.py
---
PeakOTron.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/PeakOTron.py b/PeakOTron.py
index 9d31548..56fa800 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])
--
GitLab