only save knn distances for ks we're actually using
never mind on the squaring, I was being an idiot.