@lucianopaz Thanks for your answer.
That works.
By the way, do you have any idea why one has to specify comp_shape in pm.NormalMixture? can it be figured out automatically? Also without comp_shape parameter, why does the code compile and not fail?