The docs need updating! I think what changed is now pm.sample_posterior_predictive will draw one sample for each element in the trace. What you can do now is repeat the result from find_MAP, and also subsample the results from NUTS.
The docs need updating! I think what changed is now pm.sample_posterior_predictive will draw one sample for each element in the trace. What you can do now is repeat the result from find_MAP, and also subsample the results from NUTS.