keypoints_i was a bad name choice. Better do `mask = scales == scale` and apply the mask to the x, y assignements.