I think you could simply use `[]` instead of `np.ones(0)`.