Hrmm... it'd be nice if the code supported native python lists and 1-D arrays. Would it be huge effort to update? Should be able to call `np.asarray` on the input in case it isn't an `ndarray` yet.