Dirichlet Distributions and Batching in pymc3 3.9+

It seems to be a bug - could you raise an issue on Github?