TST fix doctest for k-NN