Update estimation.py

StatisKit · Jun 23, 2020 · f8e4c06 · f8e4c06
1 parent 4f0dc89
commit f8e4c06
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/py/statiskit/core/estimation.py b/src/py/statiskit/core/estimation.py
@@ -591,6 +591,8 @@ def mixture_estimation(data, algo='em', **kwargs):
         elif outcome == outcome_type.DISCRETE:
             if kwargs.pop('singular', False):
                 mapping = dict(em = MixtureSingularDistributionEMEstimation.Estimator)
+            else:
+                mapping = dict(em = DiscreteMultivariateMixtureDistributionEMEstimation.Estimator)
         elif outcome == outcome_type.CONTINUOUS:
             mapping = dict(em = ContinuousMultivariateMixtureDistributionEMEstimation.Estimator)
     else: