Np.argmax python could not be evaluated