MCMC SamplingError: Initial evaluation of model at starting point failed!

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)