Almost there:
Your last line is wrongly indexed, it should be sorted_ids[:k] (since you want up to k top nearest neighbors, while your current slice gives you last k neighbors)
Cheers
p.s. Please remove your last image, since it’s against the code of conduct.