Update knn.py

knn.py does no work when run initially