Не могу понять, почему 'ValueError: установка элемента массива с последовательностью' происходит

Question

Я получаю 'ValueError: установка элемента массива с последовательностью' о строках 2 и 4 и не могу понять, почему ...

m_v[0, int(v), int(parent)] = psi_v[0, int(v)] * get_p_e_u_v(parent, v, 0, 0) +\
                              psi_v[1, int(v)] * get_p_e_u_v(parent, v, 0, 1)
m_v[1, int(v), int(parent)] = psi_v[0, int(v)] * get_p_e_u_v(parent, v, 1, 0) +\
                              psi_v[1, int(v)] * get_p_e_u_v(parent, v, 1, 1)

def get_p_e_u_v(u, v, xu, xv):
    if xu != xv:
        return_p_e = p_e[func(int(u), int(v))]
    else:
        return_p_e = 1.0 - p_e[func(int(u), int(v))]
    return return_p_e

fun c возвращает int в диапазон (0, len (p_e))