Convert point_logps values to arrays

When compiling to JAX the returned object is a DeviceArray, which causes problems downstream in check_start_vals
pymc-devs · Nov 27, 2021 · f6f1a8e · f6f1a8e
1 parent 01ed358
commit f6f1a8e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymc/model.py b/pymc/model.py
@@ -1628,7 +1628,9 @@ def point_logps(self, point=None, round_vals=2):
         return Series(
             {
                 rv.name: np.round(
-                    self.fn(logpt_sum(rv, getattr(rv.tag, "observations", None)))(point),
+                    np.asarray(
+                        self.fn(logpt_sum(rv, getattr(rv.tag, "observations", None)))(point)
+                    ),
                     round_vals,
                 )
                 for rv in self.basic_RVs