fixup

uxlfoundation · Nov 19, 2024 · 4c87853 · 4c87853
1 parent 2a7b8b8
commit 4c87853
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onedal/neighbors/neighbors.py b/onedal/neighbors/neighbors.py
@@ -620,7 +620,7 @@ def _onedal_fit(self, X, y, queue):
         if gpu_device:
             return self.train(policy, params, *to_table(X, y)).model
         else:
-            return self.train_search(policy, params, *to_table(X, y)).model
+            return self.train_search(policy, params, to_table(X)).model
 
     def _onedal_predict(self, model, X, params, queue):
         assert self._onedal_model is not None, "Model is not trained"