kNN pipeline, doesn't work