corrected lnp

Becksteinlab · Jan 23, 2024 · ceead80 · ceead80
1 parent 4cb6f23
commit ceead80
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/basicrta/functions.py b/basicrta/functions.py
@@ -164,9 +164,9 @@ def run(self):
 
             # Compute log posterior           
             lnp[j] = np.log(tmp.take(s)).sum()+\
-                     np.log(mcweights[j][uniqs]).sum()-\
-                     (mcrates[j][uniqs]*rhypers[uniqs, 1]).sum()+\
-                     np.log(mcweights[j][uniqs]**(whypers[uniqs]-1)).sum()
+                     np.log(mcweights[j]).sum()-\
+                     (mcrates[j]*rhypers[:, 1]).sum()+\
+                     np.log(mcweights[j]**(whypers-1)).sum()
 
             Ns[j][:] = np.array([len(inds[i]) for i in range(ncomp)])
             Ts = np.array([x[inds[i]].sum() for i in range(ncomp)])