ensure p_d is an array

aloctavodia · aloctavodia · commit abe70aa1cd67 · 2025-03-10T09:15:17.000+02:00
diff --git a/pymc_bart/utils.py b/pymc_bart/utils.py
@@ -254,7 +254,7 @@ def identity(x):
             )
 
         new_x = fake_X[:, var]
-        p_d = func(y_pred)
+        p_d = func(np.array(y_pred))
 
         for s_i in range(shape):
             if centered:
