Update KNN.py

fix a missmatch