The original implementation accepted an array which infact was just an integer cast as np.ndarray for each element of which (in this case 1), m was calculated which in turn was used to calculate the distance, which again was an np.ndarray.
I've removed an unnecessary variable size and the implementation now directly uses the 'index' provided. The results are identical as the original implementation.