Move CustomDist logic to dedicated module and docs section

.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

docs/source/api/distributions.rst

@@ -14,6 +14,7 @@ Distributions
    distributions/timeseries
    distributions/truncated
    distributions/censored
+   distributions/custom
    distributions/simulator
    distributions/transforms
    distributions/utilities

docs/source/api/distributions/custom.rst

@@ -0,0 +1,20 @@
+**********
+CustomDist
+**********
+
+..
+   Manually follow the template in _templates/distribution.rst.
+   If at any point, multiple objects are listed here,
+   the pattern should instead be modified to that of the
+   other API files such as api/distributions/continuous.rst
+
+.. currentmodule:: pymc
+
+.. autoclass:: CustomDist
+
+   .. rubric:: Methods
+
+   .. autosummary::
+      :toctree: classmethods
+
+      CustomDist.dist

docs/source/api/distributions/utilities.rst

@@ -7,7 +7,6 @@ Distribution utilities
    :toctree: generated/
 
     Continuous
-    CustomDist
     Discrete
     Distribution
     SymbolicRandomVariable

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,