Can you please update to `iterable of 1-D array_like` and ensure that the code works with lists of lists and lists of `np.array`s?