From eadb1d766f848a695a80a9029edde16202dc4a4b Mon Sep 17 00:00:00 2001 From: "Kruglov, Oleg" Date: Wed, 28 Aug 2024 09:02:51 -0700 Subject: [PATCH] Removed support_usm_nd_array --- doc/sources/quick-start.rst | 3 ++- onedal/spmd/basic_statistics/incremental_basic_statistics.py | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/doc/sources/quick-start.rst b/doc/sources/quick-start.rst index 07fa8b4fe6..70dc4603d7 100644 --- a/doc/sources/quick-start.rst +++ b/doc/sources/quick-start.rst @@ -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]]) diff --git a/onedal/spmd/basic_statistics/incremental_basic_statistics.py b/onedal/spmd/basic_statistics/incremental_basic_statistics.py index 6b686cf94c..04c378d2e7 100644 --- a/onedal/spmd/basic_statistics/incremental_basic_statistics.py +++ b/onedal/spmd/basic_statistics/incremental_basic_statistics.py @@ -16,7 +16,6 @@ from daal4py.sklearn._utils import get_dtype -from ..._device_offload import support_usm_ndarray from ...basic_statistics import ( IncrementalBasicStatistics as IncrementalBasicStatistics_nonSPMD, ) @@ -30,7 +29,6 @@ def _reset(self): IncrementalBasicStatistics_nonSPMD, self )._get_backend("basic_statistics", None, "partial_compute_result") - @support_usm_ndarray() def partial_fit(self, X, weights=None, queue=None): """ Computes partial data for basic statistics