Approximate Nearest Neighbor Python