Plotting the prior likelihood vs target in pymc

pm.sample_prior_predictive adds a dummy chain dimension (that’s the 1 at the end of the shape there). You can get rid of it with .sel(chain=0), or by squeezing the array that .values returns.