I think that this part should be using sklearn's nearest neighbor search ? _Originally posted by @bthirion in https://github.com/alexisthual/fugw/pull/68#discussion_r1709691167_