Move CustomDist code to its own module

Co-authored-by: Giuseppe Cunsolo <[email protected]>
Co-authored-by: [email protected]
Co-authored-by: "Hemant19870601" <[email protected]>

Author: 3 people

.github/workflows/tests.yml

@@ -68,6 +68,7 @@ jobs:
 
           - |
             tests/distributions/test_censored.py
+            tests/distributions/test_custom.py
             tests/distributions/test_simulator.py
             tests/sampling/test_deterministic.py
             tests/sampling/test_forward.py

pymc/distributions/__init__.py

@@ -50,6 +50,7 @@
     Wald,
     Weibull,
 )
+from pymc.distributions.custom import CustomDist, DensityDist
 from pymc.distributions.discrete import (
     Bernoulli,
     BetaBinomial,
@@ -66,8 +67,6 @@
 )
 from pymc.distributions.distribution import (
     Continuous,
-    CustomDist,
-    DensityDist,
     DiracDelta,
     Discrete,
     Distribution,