From 79441d12cc87b40ab7541a13e5726e514e91e8fc Mon Sep 17 00:00:00 2001 From: Cecilia Ferrari Date: Thu, 11 Jan 2024 09:58:32 +0100 Subject: [PATCH] fixing s2_loss annotate bug --- flamedisx/lxe_blocks/s2_loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flamedisx/lxe_blocks/s2_loss.py b/flamedisx/lxe_blocks/s2_loss.py index 5f55003be..cdc0cc0fd 100644 --- a/flamedisx/lxe_blocks/s2_loss.py +++ b/flamedisx/lxe_blocks/s2_loss.py @@ -44,7 +44,7 @@ def _annotate(self, d): s2_survival_probability = self.gimme_numpy('s2_survival_p') mle = d['s2_raw' + '_mle'] = \ - (d['s2_raw_after_loss_' + '_mle'] / s2_survival_probability).clip(0, None) + (d['s2_raw_after_loss' + '_mle'] / s2_survival_probability).clip(0, None) s = d['s2_raw'] * s2_survival_probability*(1-s2_survival_probability) scale = mle**0.5 * s / s2_survival_probability