From c5dd90f4568addc4b5d16a6bf6cad7700ad2018c Mon Sep 17 00:00:00 2001 From: T Ehrhardt Date: Fri, 20 Feb 2026 15:23:23 +0100 Subject: [PATCH] assign 0th and only element to beta & trad_alpha instead of length-1 array --- pisa/stages/likelihood/generalized_llh_params.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pisa/stages/likelihood/generalized_llh_params.py b/pisa/stages/likelihood/generalized_llh_params.py index 8db5a3f20..f960843ff 100644 --- a/pisa/stages/likelihood/generalized_llh_params.py +++ b/pisa/stages/likelihood/generalized_llh_params.py @@ -237,8 +237,8 @@ def apply_function(self): # default to alphas values of PSEUDO_WEIGHT and # of beta = 1.0, which mimicks a narrow PDF # close to 0.0 - beta = np.divide(mean_w, var_z, out=np.ones(1), where=var_z!=0) - trad_alpha = np.divide(mean_w**2, var_z, out=np.ones(1)*PSEUDO_WEIGHT, where=var_z!=0) + beta = np.divide(mean_w, var_z, out=np.ones(1), where=var_z!=0)[0] + trad_alpha = np.divide(mean_w**2, var_z, out=np.ones(1)*PSEUDO_WEIGHT, where=var_z!=0)[0] alpha = (n_weights + mean_adjustment)*trad_alpha alphas_vector[index] = alpha