COSMIT: Minor opt in pairwise_distances_argmin_min

I removed `np.where` since it was unnecessary.