Convert point_logps values to arrays

ricardoV94 · twiecki · commit f6f1a8e807a0 · 2021-11-27T12:47:10.000+01:00
When compiling to JAX the returned object is a DeviceArray, which causes problems downstream in check_start_vals
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

Original file line number	Diff line number	Diff line change
`@@ -1628,7 +1628,9 @@ def point_logps(self, point=None, round_vals=2):`
`1628`	`1628`	`return Series(`
`1629`	`1629`	`{`
`1630`	`1630`	`rv.name: np.round(`
`1631`		`- self.fn(logpt_sum(rv, getattr(rv.tag, "observations", None)))(point),`
	`1631`	`+ np.asarray(`
	`1632`	`+ self.fn(logpt_sum(rv, getattr(rv.tag, "observations", None)))(point)`
	`1633`	`+ ),`
`1632`	`1634`	`round_vals,`
`1633`	`1635`	`)`
`1634`	`1636`	`for rv in self.basic_RVs`