update density dist docstring to v4 API (#5523)

nikisix · ntomasino · web-flow · commit ac2b82ee61a9 · 2022-02-25T03:41:56.000+02:00
Co-authored-by: ntomasino &lt;nick.w.tomasino@gmail.com&gt;
diff --git a/pymc/distributions/distribution.py b/pymc/distributions/distribution.py
@@ -803,8 +803,8 @@ def random(mu, rng=None, size=None):
                         observed=np.random.randn(100, 3),
                         size=(100, 3),
                     )
-                    prior = pm.sample_prior_predictive(10)['density_dist']
-                assert prior.shape == (10, 100, 3)
+                    prior = pm.sample_prior_predictive(10).prior_predictive['density_dist']
+                assert prior.shape == (1, 10, 100, 3)
 
         """
 

Original file line number	Diff line number	Diff line change
`@@ -803,8 +803,8 @@ def random(mu, rng=None, size=None):`
`803`	`803`	`observed=np.random.randn(100, 3),`
`804`	`804`	`size=(100, 3),`
`805`	`805`	`)`
`806`		`- prior = pm.sample_prior_predictive(10)['density_dist']`
`807`		`- assert prior.shape == (10, 100, 3)`
	`806`	`+ prior = pm.sample_prior_predictive(10).prior_predictive['density_dist']`
	`807`	`+ assert prior.shape == (1, 10, 100, 3)`
`808`	`808`
`809`	`809`	`"""`
`810`	`810`