Some functions no longer work because of Numpy convention For example : np.sum(np.abs(A * Omega)) It does not return a scalar but a vector It could be solved using np.linalg.norm(A * Omega, 1)