The posterior predictive should be of shape: (nchains, ndraws, nobservations). You can check by printing:
test_ppc.posterior_predictive.dims
The posterior predictive should be of shape: (nchains, ndraws, nobservations). You can check by printing:
test_ppc.posterior_predictive.dims