Error Found in knn.py

[thepranaygaur]

My code:
def get_neighbourhood(self,X_train, y_train, point, K):
pairs = zip(X_train, y_train)
pairs.sort(key = lambda pair: self.euclidean_distance(point, pair[0]))
pairs = pairs[0:K]
x = zip(*pairs[1])
return mode(x)
Your code:
def get_neighbourhood(self,X_train, y_train, point, K):
pairs = zip(X_train, y_train)
pairs.sort(key = lambda pair: self.euclidean_distance(point, pair[0]))
pairs = pairs[0:K]
return mode(zip(*pairs)[1])

Error found:
File "/home/vulcan/Documents/Python/NIDS/knn.py", line 28, in get_neighbourhood
x = zip(*pairs[1])
TypeError: zip argument #2 must support iteration

tried converting zip argument to list but not working