Update ppi.py

aangelopoulos · Jul 9, 2024 · ac99d9a · ac99d9a
1 parent 27ac724
commit ac99d9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppi_py/ppi.py b/ppi_py/ppi.py
@@ -1691,7 +1691,7 @@ def _calc_lam_glm(
     )
     var_grads_hat = var_grads_hat.reshape(d, d)
 
-    vhat = inv_hessian if coord is None else inv_hessian[coord, coord]
+    vhat = inv_hessian if coord is None else inv_hessian[coord, :]
     if optim_mode == "overall":
         num = (
             np.trace(vhat @ cov_grads @ vhat)