From 7625457493309f4a42116192b5a36bdede8a52e3 Mon Sep 17 00:00:00 2001 From: "Kruglov, Oleg" Date: Mon, 2 Sep 2024 02:37:19 -0700 Subject: [PATCH] Change naming for base class reference --- onedal/spmd/covariance/incremental_covariance.py | 10 +++++----- sklearnex/spmd/covariance/incremental_covariance.py | 4 ++-- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/onedal/spmd/covariance/incremental_covariance.py b/onedal/spmd/covariance/incremental_covariance.py index 7851ab40f6..e0840c3ac6 100644 --- a/onedal/spmd/covariance/incremental_covariance.py +++ b/onedal/spmd/covariance/incremental_covariance.py @@ -19,7 +19,7 @@ from daal4py.sklearn._utils import get_dtype from ...covariance import ( - IncrementalEmpiricalCovariance as IncrementalEmpiricalCovariance_base, + IncrementalEmpiricalCovariance as base_IncrementalEmpiricalCovariance, ) from ...datatypes import _convert_to_supported, to_table from ...utils import _check_array @@ -27,11 +27,11 @@ class IncrementalEmpiricalCovariance( - BaseEstimatorSPMD, IncrementalEmpiricalCovariance_base + BaseEstimatorSPMD, base_IncrementalEmpiricalCovariance ): def _reset(self): self._partial_result = super( - IncrementalEmpiricalCovariance_base, self + base_IncrementalEmpiricalCovariance, self )._get_backend("covariance", None, "partial_compute_result") def partial_fit(self, X, y=None, queue=None): @@ -60,7 +60,7 @@ def partial_fit(self, X, y=None, queue=None): self._queue = queue - policy = super(IncrementalEmpiricalCovariance_base, self)._get_policy(queue, X) + policy = super(base_IncrementalEmpiricalCovariance, self)._get_policy(queue, X) X = _convert_to_supported(policy, X) @@ -70,7 +70,7 @@ def partial_fit(self, X, y=None, queue=None): params = self._get_onedal_params(self._dtype) table_X = to_table(X) self._partial_result = super( - IncrementalEmpiricalCovariance_base, self + base_IncrementalEmpiricalCovariance, self )._get_backend( "covariance", None, diff --git a/sklearnex/spmd/covariance/incremental_covariance.py b/sklearnex/spmd/covariance/incremental_covariance.py index cc0a0c8f65..b784891e03 100644 --- a/sklearnex/spmd/covariance/incremental_covariance.py +++ b/sklearnex/spmd/covariance/incremental_covariance.py @@ -19,11 +19,11 @@ ) from ...covariance import ( - IncrementalEmpiricalCovariance as IncrementalEmpiricalCovariance_base, + IncrementalEmpiricalCovariance as base_IncrementalEmpiricalCovariance, ) -class IncrementalEmpiricalCovariance(IncrementalEmpiricalCovariance_base): +class IncrementalEmpiricalCovariance(base_IncrementalEmpiricalCovariance): _onedal_incremental_covariance = staticmethod( onedalSPMD_IncrementalEmpiricalCovariance )