Removed support_usm_nd_array

uxlfoundation · Aug 28, 2024 · a9ab64e · a9ab64e
1 parent be2f272
commit a9ab64e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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