From 252c8a3d298b39f873f86318b7d53bdab3ff56ac Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Mon, 28 Jul 2025 20:23:59 +0100
Subject: [PATCH] Refactor KMeans module to use inbuilt random

---
 src/compute/models/k_means.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/compute/models/k_means.rs b/src/compute/models/k_means.rs
index 83412e3..0766075 100644
--- a/src/compute/models/k_means.rs
+++ b/src/compute/models/k_means.rs
@@ -1,7 +1,6 @@
 use crate::compute::stats::mean_vertical;
 use crate::matrix::Matrix;
-use rand::rng;
-use rand::seq::SliceRandom;
+use crate::random::prelude::*;
 
 pub struct KMeans {
     pub centroids: Matrix<f64>, // (k, n_features)
@@ -193,7 +192,11 @@ mod tests {
                                 break;
                             }
                         }
-                        assert!(matches_data_point, "Centroid {} (empty cluster) does not match any data point", c);
+                        assert!(
+                            matches_data_point,
+                            "Centroid {} (empty cluster) does not match any data point",
+                            c
+                        );
                     }
                 }
                 break;
@@ -360,5 +363,4 @@ mod tests {
         assert_eq!(predicted_label.len(), 1);
         assert!(predicted_label[0] < k);
     }
-
 }