fix: fix tupleerror

Author: yuejiaointel

sklearnex/neighbors/knn_classification.py

@@ -240,7 +240,8 @@ def _save_attributes(self):
         self.classes_ = self._onedal_estimator.classes_
         self.n_features_in_ = self._onedal_estimator.n_features_in_
         self.n_samples_fit_ = self._onedal_estimator.n_samples_fit_
-        self._fit_X = self._onedal_estimator._fit_X
+        fit_x = self._onedal_estimator._fit_X
+        self._fit_X = fit_x[0] if isinstance(fit_x, tuple) else fit_x
         self._y = self._onedal_estimator._y
         self._fit_method = self._onedal_estimator._fit_method
         self.outputs_2d_ = self._onedal_estimator.outputs_2d_

sklearnex/neighbors/knn_regression.py

@@ -201,7 +201,8 @@ def _onedal_score(self, X, y, sample_weight=None, queue=None):
     def _save_attributes(self):
         self.n_features_in_ = self._onedal_estimator.n_features_in_
         self.n_samples_fit_ = self._onedal_estimator.n_samples_fit_
-        self._fit_X = self._onedal_estimator._fit_X
+        fit_x = self._onedal_estimator._fit_X
+        self._fit_X = fit_x[0] if isinstance(fit_x, tuple) else fit_x
         self._y = self._onedal_estimator._y
         self._fit_method = self._onedal_estimator._fit_method
         self._tree = self._onedal_estimator._tree

sklearnex/neighbors/knn_unsupervised.py

@@ -186,7 +186,8 @@ def _save_attributes(self):
         self.classes_ = self._onedal_estimator.classes_
         self.n_features_in_ = self._onedal_estimator.n_features_in_
         self.n_samples_fit_ = self._onedal_estimator.n_samples_fit_
-        self._fit_X = self._onedal_estimator._fit_X
+        fit_x = self._onedal_estimator._fit_X
+        self._fit_X = fit_x[0] if isinstance(fit_x, tuple) else fit_x
         self._fit_method = self._onedal_estimator._fit_method
         self._tree = self._onedal_estimator._tree