Merge 12a72317e4 into 11330e464b

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-6);
-        assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-6);
+        assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-9);
+        assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-9);
     }
 
     #[test]