see if this works

uxlfoundation · Dec 6, 2024 · b64f962 · b64f962
1 parent c47daf3
commit b64f962
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 5 deletions.
diff --git a/sklearnex/basic_statistics/basic_statistics.py b/sklearnex/basic_statistics/basic_statistics.py
@@ -26,7 +26,8 @@
 from onedal.basic_statistics import BasicStatistics as onedal_BasicStatistics
 
 from .._device_offload import dispatch
-from .._utils import IntelEstimator, PatchingConditionsChain
+from .._utils import PatchingConditionsChain
+from ..base import IntelEstimator
 
 if sklearn_check_version("1.6"):
     from sklearn.utils.validation import validate_data

diff --git a/sklearnex/basic_statistics/incremental_basic_statistics.py b/sklearnex/basic_statistics/incremental_basic_statistics.py
@@ -26,7 +26,8 @@
 )
 
 from .._device_offload import dispatch
-from .._utils import IntelEstimator, PatchingConditionsChain
+from .._utils import PatchingConditionsChain
+from ..base import IntelEstimator
 
 if sklearn_check_version("1.2"):
     from sklearn.utils._param_validation import Interval, StrOptions

diff --git a/sklearnex/covariance/incremental_covariance.py b/sklearnex/covariance/incremental_covariance.py
@@ -33,7 +33,8 @@
 from sklearnex import config_context
 
 from .._device_offload import dispatch, wrap_output_data
-from .._utils import IntelEstimator, PatchingConditionsChain, register_hyperparameters
+from .._utils import PatchingConditionsChain, register_hyperparameters
+from ..base import IntelEstimator
 from ..metrics import pairwise_distances
 from ..utils._array_api import get_namespace
 

diff --git a/sklearnex/linear_model/incremental_linear.py b/sklearnex/linear_model/incremental_linear.py
@@ -40,7 +40,8 @@
 from onedal.common.hyperparameters import get_hyperparameters
 
 from .._device_offload import dispatch, wrap_output_data
-from .._utils import IntelEstimator, PatchingConditionsChain, register_hyperparameters
+from .._utils import PatchingConditionsChain, register_hyperparameters
+from ..base import IntelEstimator
 
 
 @register_hyperparameters(

diff --git a/sklearnex/linear_model/incremental_ridge.py b/sklearnex/linear_model/incremental_ridge.py
@@ -33,6 +33,7 @@
 
 from .._device_offload import dispatch, wrap_output_data
 from .._utils import PatchingConditionsChain
+from ..base import IntelEstimator
 
 if sklearn_check_version("1.6"):
     from sklearn.utils.validation import validate_data
@@ -43,7 +44,7 @@
 @control_n_jobs(
     decorated_methods=["fit", "partial_fit", "predict", "score", "_onedal_finalize_fit"]
 )
-class IncrementalRidge(MultiOutputMixin, RegressorMixin, BaseEstimator):
+class IncrementalRidge(IntelEstimator, MultiOutputMixin, RegressorMixin, BaseEstimator):
     """
     Incremental estimator for Ridge Regression.
     Allows to train Ridge Regression if data is splitted into batches.