`pm.sample_prior_predictive()` fails with incorrect size info passed to the `rng_fn` function of the RandomVariable

Thank you so much, @ricardoV94!! This is exactly what I was looking for. Really appreciate the tip!