From ac2b82ee61a95493df6023e63c3c89cb882d0640 Mon Sep 17 00:00:00 2001
From: niki six <nicolai.tesela@gmail.com>
Date: Thu, 24 Feb 2022 17:41:56 -0800
Subject: [PATCH] update density dist docstring to v4 API (#5523)

Co-authored-by: ntomasino <nick.w.tomasino@gmail.com>
---
 pymc/distributions/distribution.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pymc/distributions/distribution.py b/pymc/distributions/distribution.py
index 34bb0a71726..89f37a13749 100644
--- 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)
 
         """