Skip to content

Commit

Permalink
Remove support_usm_ndarray
Browse files Browse the repository at this point in the history
  • Loading branch information
olegkkruglov committed Aug 28, 2024
1 parent 1618e09 commit 805e08d
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 7 deletions.
3 changes: 2 additions & 1 deletion doc/sources/quick-start.rst
Original file line number Diff line number Diff line change
Expand Up @@ -79,12 +79,13 @@ This example shows how to patch Intel(R) extension for Scikit-Learn by modifing
:caption: Example: Drop-In Patching
import numpy as np
from sklearnex import patch_sklearn
patch_sklearn()
# You need to re-import scikit-learn algorithms after the patch
from sklearn.cluster import KMeans
# The use of the original Scikit-learn is not changed
X = np.array([[1, 2], [1, 4], [1, 0],
[10, 2], [10, 4], [10, 0]])
Expand Down
6 changes: 0 additions & 6 deletions onedal/spmd/linear_model/incremental_linear_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@

from daal4py.sklearn._utils import get_dtype

from ..._device_offload import support_usm_ndarray
from ...common.hyperparameters import get_hyperparameters
from ...datatypes import _convert_to_supported, to_table
from ...linear_model import (
Expand All @@ -34,7 +33,6 @@ def _reset(self):
IncrementalLinearRegression_nonSPMD, self
)._get_backend("linear_model", "regression", "partial_train_result")

@support_usm_ndarray()
def partial_fit(self, X, y, queue=None):
"""
Computes partial data for linear regression
Expand Down Expand Up @@ -91,7 +89,3 @@ def partial_fit(self, X, y, queue=None):
self._partial_result = module.partial_train(
policy, self._params, self._partial_result, X_table, y_table
)

@support_usm_ndarray()
def finalize_fit(self, queue=None):
return super().finalize_fit(queue=queue)

0 comments on commit 805e08d

Please sign in to comment.