From ab6d5f9f8ff72cd77ed2d2a610fc38bac72972a0 Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Sun, 6 Jul 2025 19:17:09 +0100
Subject: [PATCH] Refactor test module imports in LinReg to improve clarity

---
 src/compute/models/linreg.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/compute/models/linreg.rs b/src/compute/models/linreg.rs
index 5add9f9..ba76197 100644
--- a/src/compute/models/linreg.rs
+++ b/src/compute/models/linreg.rs
@@ -34,10 +34,10 @@ impl LinReg {
     }
 }
 
+#[cfg(test)]
 mod tests {
 
-    use super::LinReg;
-    use crate::matrix::{Matrix};
+    use super::*;
 
     #[test]
     fn test_linreg_fit_predict() {