Skip to content

Commit c93711a

Browse files
rowan-schaeferRowan Schaefer
and
Rowan Schaefer
authored
issue6004 fixed example in docstring for set_data (#6028)
Co-authored-by: Rowan Schaefer <[email protected]>
1 parent 36c5fb9 commit c93711a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1870,7 +1870,7 @@ def set_data(new_data, model=None, *, coords=None):
18701870
>>> with model:
18711871
... pm.set_data({'x': [5., 6., 9.]})
18721872
... y_test = pm.sample_posterior_predictive(idata)
1873-
>>> y_test['obs'].mean(axis=0)
1873+
>>> y_test.posterior_predictive['obs'].mean(('chain', 'draw'))
18741874
array([4.6088569 , 5.54128318, 8.32953844])
18751875
"""
18761876
model = modelcontext(model)

0 commit comments

Comments
 (0)