pub fn compute_partition<T: Float + L2 + Dot>( centroids: &[T], vector: &[T], distance_type: DistanceType, ) -> Option<u32>