Sorry forget about this comment, didn't see the np.asarray step.