Consistency check in Discrete.merge method

This commit is contained in:
Paul Romano 2022-05-11 14:47:07 -05:00
parent cf665d076f
commit 178adb0843

View file

@ -174,6 +174,9 @@ class Discrete(Univariate):
Combined discrete distribution
"""
if len(dists) != len(probs):
raise ValueError("Number of distributions and probabilities must match.")
# Combine distributions accounting for duplicate x values
x_merged = set()
p_merged = defaultdict(float)