Export DirichletMultinomial in pymc3.distributions

As suggested in #3639 (comment) Also see: #3639 (comment) but this seems to be part of a broader discussion.
pymc-devs · Dec 7, 2019 · e932ae6 · e932ae6
1 parent aed90fa
commit e932ae6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pymc3/distributions/__init__.py b/pymc3/distributions/__init__.py
@@ -69,6 +69,7 @@
 from .multivariate import MvStudentT
 from .multivariate import Dirichlet
 from .multivariate import Multinomial
+from .multivariate import DirichletMultinomial
 from .multivariate import Wishart
 from .multivariate import WishartBartlett
 from .multivariate import LKJCholeskyCov
@@ -133,6 +134,7 @@
            'MvStudentT',
            'Dirichlet',
            'Multinomial',
+           'DirichletMultinomial',
            'Wishart',
            'WishartBartlett',
            'LKJCholeskyCov',