I think you should call `array` `y`, and include `y = np.asarray(y)` in case someone calls this and gets weird results when passing a list.