Problem with pm.Categorical

lol. BTW you might want to use a Dirichlet prior so it generates unit vector p