Skip to content

Commit 970db18

Browse files
tomicaprettoricardoV94
authored andcommitted
Move distributions to more consistent places
ZeroInflated to mixture.py DiracDelta to distribution.py
1 parent 83e3545 commit 970db18

File tree

11 files changed

+579
-561
lines changed

11 files changed

+579
-561
lines changed

docs/source/api/distributions/discrete.rst

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ Discrete
1010
BetaBinomial
1111
Binomial
1212
Categorical
13-
DiracDelta
1413
DiscreteUniform
1514
DiscreteWeibull
1615
Geometric
@@ -19,6 +18,3 @@ Discrete
1918
OrderedLogistic
2019
OrderedProbit
2120
Poisson
22-
ZeroInflatedBinomial
23-
ZeroInflatedNegativeBinomial
24-
ZeroInflatedPoisson

docs/source/api/distributions/mixture.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,6 @@ Mixture
88

99
Mixture
1010
NormalMixture
11+
ZeroInflatedBinomial
12+
ZeroInflatedNegativeBinomial
13+
ZeroInflatedPoisson

docs/source/api/distributions/utilities.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,4 @@ Distribution utilities
1111
Discrete
1212
Distribution
1313
SymbolicRandomVariable
14+
DiracDelta

pymc/distributions/__init__.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,6 @@
5555
BetaBinomial,
5656
Binomial,
5757
Categorical,
58-
DiracDelta,
5958
DiscreteUniform,
6059
DiscreteWeibull,
6160
Geometric,
@@ -64,19 +63,23 @@
6463
OrderedLogistic,
6564
OrderedProbit,
6665
Poisson,
67-
ZeroInflatedBinomial,
68-
ZeroInflatedNegativeBinomial,
69-
ZeroInflatedPoisson,
7066
)
7167
from pymc.distributions.distribution import (
7268
Continuous,
7369
CustomDist,
7470
DensityDist,
71+
DiracDelta,
7572
Discrete,
7673
Distribution,
7774
SymbolicRandomVariable,
7875
)
79-
from pymc.distributions.mixture import Mixture, NormalMixture
76+
from pymc.distributions.mixture import (
77+
Mixture,
78+
NormalMixture,
79+
ZeroInflatedBinomial,
80+
ZeroInflatedNegativeBinomial,
81+
ZeroInflatedPoisson,
82+
)
8083
from pymc.distributions.multivariate import (
8184
CAR,
8285
Dirichlet,

0 commit comments

Comments
 (0)