Change KMeansMiniBatch to use takeSample() instead of sample().  Update tests accordingly.