Why use `np.arange(1, 3)` here but `[1, 2]` above?