I fixed it by chagning the apply function into:
def apply(self, rv):
return pm.distributions.transforms.TransformedDistribution.dist(rv.distribution, self)
I fixed it by chagning the apply function into:
def apply(self, rv):
return pm.distributions.transforms.TransformedDistribution.dist(rv.distribution, self)