That’s a neat solution, you can do it without the class as well
truncated_poisson = pm.Truncated.dist(pm.Poisson.dist(), lower=0, upper=upper_limit)
zero_inflation = pm.DiracDelta(0)
truncated_zip = pm.Mixture("truncated_zip", w=..., comp_dists([zero_inflation, truncated_poisson]))