Automatic imputation - array dimension problem

I recommend reviewing the discussion at these threads:

In particular, the first thread recommends using trace = pm.sample(return_inferencedata=False, compute_convergence_checks=False) so that the postprocessing doesn’t execute.

2 Likes