in src/utils.py [0:0]
def cum_sum_prob(prob_dict):
"""Calculate cumulative probability from a list of probabilities"""
if not math.isclose(sum(prob_dict.values()), 1, rel_tol=1e-3):
ValueError('Input probabilities do not sum to 1.')
out = []
cur_sum = 0
for k, v in prob_dict.items():
cur_sum += v
out.append((k, cur_sum))
return out