update density dist docstring to v4 API (#5523)

Co-authored-by: ntomasino <nick.w.tomasino@gmail.com>
pymc-devs · Feb 25, 2022 · ac2b82e · ac2b82e
1 parent 4dade6d
commit ac2b82e
Showing 1 changed file with 2 additions and 2 deletions.
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)
 
         """