diff --git a/sklearnex/covariance/incremental_covariance.py b/sklearnex/covariance/incremental_covariance.py index 582b5316dc..3ceac4018f 100644 --- a/sklearnex/covariance/incremental_covariance.py +++ b/sklearnex/covariance/incremental_covariance.py @@ -31,7 +31,6 @@ IncrementalEmpiricalCovariance as onedal_IncrementalEmpiricalCovariance, ) from onedal.utils._array_api import _is_numpy_namespace -from sklearnex import config_context from .._device_offload import dispatch, wrap_output_data from .._utils import IntelEstimator, PatchingConditionsChain, register_hyperparameters diff --git a/sklearnex/preview/covariance/covariance.py b/sklearnex/preview/covariance/covariance.py index c75f1bee0b..1f98d74ac8 100644 --- a/sklearnex/preview/covariance/covariance.py +++ b/sklearnex/preview/covariance/covariance.py @@ -24,6 +24,7 @@ from daal4py.sklearn._utils import daal_check_version, sklearn_check_version from onedal.common.hyperparameters import get_hyperparameters from onedal.covariance import EmpiricalCovariance as onedal_EmpiricalCovariance +from onedal.utils._array_api import _is_numpy_namespace from sklearnex import config_context from sklearnex.metrics import pairwise_distances