So then should the docstring be changed to say `list` or `np.array`?