From c24eb4a08c2585668ec4389d24df5848fab922b6 Mon Sep 17 00:00:00 2001 From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com> Date: Sat, 12 Jul 2025 01:47:40 +0100 Subject: [PATCH] Relax assertion tolerance in KMeans tests to align with algorithm's convergence criteria --- src/compute/models/k_means.rs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/compute/models/k_means.rs b/src/compute/models/k_means.rs index fe67e8e..a92e414 100644 --- a/src/compute/models/k_means.rs +++ b/src/compute/models/k_means.rs @@ -265,8 +265,9 @@ mod tests { let expected_centroid_x = x.column(0).iter().sum::() / x.rows() as f64; let expected_centroid_y = x.column(1).iter().sum::() / x.rows() as f64; - assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-9); - assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-9); + // 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); } #[test]