Como evitar erros de valor ao usar numpy.random.multinomial?

Quando eu uso esse gerador aleatório:numpy.random.multinomial, Continuo recebendo:

ValueError: sum(pvals[:-1]) > 1.0

Estou sempre passando a saída dessa função softmax:

def softmax(w, t = 1.0):
    e = numpy.exp(numpy.array(w) / t)
    dist = e / np.sum(e)
    return dist

exceto agora que estou recebendo esse erro, também adicionei isso ao parâmetro (pvals):

while numpy.sum(pvals) > 1:
    pvals /= (1+1e-5)

mas isso não resolveu. Qual é o caminho certo para evitar esse erro?

EDIT: aqui está a função que inclui este código

def get_MDN_prediction(vec):
    coeffs = vec[::3]
    means = vec[1::3]
    stds = np.log(1+np.exp(vec[2::3]))
    stds = np.maximum(stds, min_std)
    coe = softmax(coeffs)
    while np.sum(coe) > 1-1e-9:
        coe /= (1+1e-5)
    coeff = unhot(np.random.multinomial(1, coe))
    return np.random.normal(means[coeff], stds[coeff])