Move distributions to more consistent places

ZeroInflated to mixture.py DiracDelta to distribution.py
pymc-devs · May 10, 2023 · 970db18 · 970db18
1 parent 83e3545
commit 970db18
Show file tree

Hide file tree

Showing 11 changed files with 579 additions and 561 deletions.
diff --git a/docs/source/api/distributions/discrete.rst b/docs/source/api/distributions/discrete.rst
@@ -10,7 +10,6 @@ Discrete
    BetaBinomial
    Binomial
    Categorical
-   DiracDelta
    DiscreteUniform
    DiscreteWeibull
    Geometric
@@ -19,6 +18,3 @@ Discrete
    OrderedLogistic
    OrderedProbit
    Poisson
-   ZeroInflatedBinomial
-   ZeroInflatedNegativeBinomial
-   ZeroInflatedPoisson
diff --git a/docs/source/api/distributions/mixture.rst b/docs/source/api/distributions/mixture.rst
@@ -8,3 +8,6 @@ Mixture
 
    Mixture
    NormalMixture
+   ZeroInflatedBinomial
+   ZeroInflatedNegativeBinomial
+   ZeroInflatedPoisson
diff --git a/docs/source/api/distributions/utilities.rst b/docs/source/api/distributions/utilities.rst
@@ -11,3 +11,4 @@ Distribution utilities
     Discrete
     Distribution
     SymbolicRandomVariable
+    DiracDelta
diff --git a/pymc/distributions/__init__.py b/pymc/distributions/__init__.py
@@ -55,7 +55,6 @@
     BetaBinomial,
     Binomial,
     Categorical,
-    DiracDelta,
     DiscreteUniform,
     DiscreteWeibull,
     Geometric,
@@ -64,19 +63,23 @@
     OrderedLogistic,
     OrderedProbit,
     Poisson,
-    ZeroInflatedBinomial,
-    ZeroInflatedNegativeBinomial,
-    ZeroInflatedPoisson,
 )
 from pymc.distributions.distribution import (
     Continuous,
     CustomDist,
     DensityDist,
+    DiracDelta,
     Discrete,
     Distribution,
     SymbolicRandomVariable,
 )
-from pymc.distributions.mixture import Mixture, NormalMixture
+from pymc.distributions.mixture import (
+    Mixture,
+    NormalMixture,
+    ZeroInflatedBinomial,
+    ZeroInflatedNegativeBinomial,
+    ZeroInflatedPoisson,
+)
 from pymc.distributions.multivariate import (
     CAR,
     Dirichlet,