"""This is a small piece of functionality extracted from 
the Scikit-learn codebase. See: https://github.com/scikit-learn/scikit-learn/blob/main/sklearn/impute/_knn.py#L262"""