fix: removed check_array from onedal4py ridge to speed up 5-6 times

intel · Nov 12, 2024 · 64bbf53 · 64bbf53
1 parent 4887b8f
commit 64bbf53
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/onedal/linear_model/linear_model.py b/onedal/linear_model/linear_model.py
@@ -293,15 +293,15 @@ def fit(self, X, y, queue=None):
         """
         module = self._get_backend("linear_model", "regression")
 
-        X = _check_array(
-            X,
-            dtype=[np.float64, np.float32],
-            force_all_finite=False,
-            ensure_2d=False,
-            copy=self.copy_X,
-        )
+        if not isinstance(X, np.ndarray):
+            X = np.asarray(X)
+
+        dtype = get_dtype(X)
+        if dtype not in [np.float32, np.float64]:
+            dtype = np.float64
+            X = X.astype(dtype, copy=self.copy_X)
 
-        y = np.asarray(y).astype(dtype=get_dtype(X))
+        y = np.asarray(y).astype(dtype=dtype)
 
         X, y = _check_X_y(X, y, force_all_finite=False, accept_2d_y=True)