Merge 12a72317e4ed93ee54fce8ea0aff83b613067605 into 11330e464ba3a7f08aaf73bc918281472c503b1d

src/compute/models/k_means.rs

@@ -265,9 +265,8 @@ mod tests {
         let expected_centroid_x = x.column(0).iter().sum::<f64>() / x.rows() as f64;
         let expected_centroid_y = x.column(1).iter().sum::<f64>() / x.rows() as f64;
 
-        // Relax the assertion tolerance to match the algorithm's convergence tolerance
+        assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-9);
-        assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-6);
+        assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-9);
-        assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-6);
     }
 
     #[test]