Adding a generator to a generator on this line & in line 126 are why Travis is complaining. `np.arange` should do the trick.