Change naming for base class reference

uxlfoundation · Sep 2, 2024 · 387fd52 · 387fd52
1 parent fa1cc04
commit 387fd52
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/onedal/spmd/linear_model/incremental_linear_model.py b/onedal/spmd/linear_model/incremental_linear_model.py
@@ -21,15 +21,15 @@
 from ...common.hyperparameters import get_hyperparameters
 from ...datatypes import _convert_to_supported, to_table
 from ...linear_model import (
-    IncrementalLinearRegression as IncrementalLinearRegression_base,
+    IncrementalLinearRegression as base_IncrementalLinearRegression,
 )
 from ...utils import _check_X_y, _num_features
 from .._base import BaseEstimatorSPMD
 
 
-class IncrementalLinearRegression(BaseEstimatorSPMD, IncrementalLinearRegression_base):
+class IncrementalLinearRegression(BaseEstimatorSPMD, base_IncrementalLinearRegression):
     def _reset(self):
-        self._partial_result = super(IncrementalLinearRegression_base, self)._get_backend(
+        self._partial_result = super(base_IncrementalLinearRegression, self)._get_backend(
             "linear_model", "regression", "partial_train_result"
         )
 
@@ -54,12 +54,12 @@ def partial_fit(self, X, y, queue=None):
         self : object
             Returns the instance itself.
         """
-        module = super(IncrementalLinearRegression_base, self)._get_backend(
+        module = super(base_IncrementalLinearRegression, self)._get_backend(
             "linear_model", "regression"
         )
 
         self._queue = queue
-        policy = super(IncrementalLinearRegression_base, self)._get_policy(queue, X)
+        policy = super(base_IncrementalLinearRegression, self)._get_policy(queue, X)
 
         X, y = _convert_to_supported(policy, X, y)
 

diff --git a/sklearnex/spmd/linear_model/incremental_linear_model.py b/sklearnex/spmd/linear_model/incremental_linear_model.py
@@ -20,9 +20,9 @@
 )
 
 from ...linear_model import (
-    IncrementalLinearRegression as IncrementalLinearRegression_base,
+    IncrementalLinearRegression as base_IncrementalLinearRegression,
 )
 
 
-class IncrementalLinearRegression(IncrementalLinearRegression_base):
+class IncrementalLinearRegression(base_IncrementalLinearRegression):
     _onedal_incremental_linear = staticmethod(onedalSPMD_IncrementalLinearRegression)