write a K-Nearest-Neighbor code on python

import numpy as np # Write your own implementation of KNN. Your code should pass tests in the cell below. class KNN: def __init__(self, k=3): """ Method is called when an object is created """ self.k = k self.X_train = None self.y_train = None def fit(self, X_train, y_train): """ Method is called to train (fit) the model """ # Your code here return def predict(self, X_test) -> np.ndarray: """ Method is called to predict labels for new data """ # Your code here return predictions # predictions should be in format of Numpy array