class DirichletMixture:
def __new__(cls, name, w, p=None, comp_shape=(), **kwargs):
return pm.Mixture(name, w, pm.Dirichlet.dist(a=p, shape=comp_shape), **kwargs)
@classmethod
def dist(cls, w, p=None, comp_shape=(), **kwargs):
return pm.Mixture.dist(w, pm.Dirichlet.dist(a=p, shape=comp_shape), **kwargs)