pm.DensityDist issue (pymc3 vs pymc5)

The API has changed, you can check the new documentation here: pymc.CustomDist — PyMC dev documentation