How can i sample posteriors from training-data, run posterior predictive on testing data (with complex factor graph)

hiya, could you please provide the code you used for plotting the predictions, seems like these plots changed a bit since pymc3, did you manually sort it(.mean() per chain to get it from 3d-2d and then plotting) out or do we have access to a az function that produces this directly, couldnt find any in the docs beside az.plot_ppc but its a bit different…