scikit-learn-intelex 2025.0.1__py310-none-win_amd64.whl → 2025.2.0__py310-none-win_amd64.whl

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/covariance.py

@@ -18,7 +18,7 @@ import warnings
 
 import numpy as np
 from scipy import sparse as sp
-from sklearn.covariance import EmpiricalCovariance as sklearn_EmpiricalCovariance
+from sklearn.covariance import EmpiricalCovariance as _sklearn_EmpiricalCovariance
 from sklearn.utils import check_array
 
 from daal4py.sklearn._n_jobs_support import control_n_jobs
@@ -31,15 +31,20 @@ from sklearnex.metrics import pairwise_distances
 from ..._device_offload import dispatch, wrap_output_data
 from ..._utils import PatchingConditionsChain, register_hyperparameters
 
+if sklearn_check_version("1.6"):
+    from sklearn.utils.validation import validate_data
+else:
+    validate_data = _sklearn_EmpiricalCovariance._validate_data
+
 
 @register_hyperparameters({"fit": get_hyperparameters("covariance", "compute")})
 @control_n_jobs(decorated_methods=["fit", "mahalanobis"])
-class EmpiricalCovariance(sklearn_EmpiricalCovariance):
-    __doc__ = sklearn_EmpiricalCovariance.__doc__
+class EmpiricalCovariance(_sklearn_EmpiricalCovariance):
+    __doc__ = _sklearn_EmpiricalCovariance.__doc__
 
     if sklearn_check_version("1.2"):
         _parameter_constraints: dict = {
-            **sklearn_EmpiricalCovariance._parameter_constraints,
+            **_sklearn_EmpiricalCovariance._parameter_constraints,
         }
 
     def _save_attributes(self):
@@ -91,7 +96,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
         if sklearn_check_version("1.2"):
             self._validate_params()
         if sklearn_check_version("0.23"):
-            X = self._validate_data(X, force_all_finite=False)
+            X = validate_data(self, X, force_all_finite=False)
         else:
             X = check_array(X, force_all_finite=False)
 
@@ -100,7 +105,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
             "fit",
             {
                 "onedal": self.__class__._onedal_fit,
-                "sklearn": sklearn_EmpiricalCovariance.fit,
+                "sklearn": _sklearn_EmpiricalCovariance.fit,
             },
             X,
         )
@@ -111,7 +116,7 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
     @wrap_output_data
     def mahalanobis(self, X):
         if sklearn_check_version("1.0"):
-            X = self._validate_data(X, reset=False)
+            X = validate_data(self, X, reset=False)
         else:
             X = check_array(X)
 
@@ -124,10 +129,10 @@ class EmpiricalCovariance(sklearn_EmpiricalCovariance):
 
         return np.reshape(dist, (len(X),)) ** 2
 
-    error_norm = wrap_output_data(sklearn_EmpiricalCovariance.error_norm)
-    score = wrap_output_data(sklearn_EmpiricalCovariance.score)
+    error_norm = wrap_output_data(_sklearn_EmpiricalCovariance.error_norm)
+    score = wrap_output_data(_sklearn_EmpiricalCovariance.score)
 
-    fit.__doc__ = sklearn_EmpiricalCovariance.fit.__doc__
-    mahalanobis.__doc__ = sklearn_EmpiricalCovariance.mahalanobis
-    error_norm.__doc__ = sklearn_EmpiricalCovariance.error_norm.__doc__
-    score.__doc__ = sklearn_EmpiricalCovariance.score.__doc__
+    fit.__doc__ = _sklearn_EmpiricalCovariance.fit.__doc__
+    mahalanobis.__doc__ = _sklearn_EmpiricalCovariance.mahalanobis
+    error_norm.__doc__ = _sklearn_EmpiricalCovariance.error_norm.__doc__
+    score.__doc__ = _sklearn_EmpiricalCovariance.score.__doc__

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/preview/covariance/tests/test_covariance.py

@@ -36,7 +36,7 @@ def test_sklearnex_import_covariance(dataframe, queue, macro_block, assume_cente
     X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
     empcov = EmpiricalCovariance(assume_centered=assume_centered)
     if daal_check_version((2024, "P", 0)) and macro_block is not None:
-        hparams = empcov.get_hyperparameters("fit")
+        hparams = EmpiricalCovariance.get_hyperparameters("fit")
         hparams.cpu_macro_block = macro_block
     result = empcov.fit(X)
 

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/preview/decomposition/incremental_pca.py

@@ -15,7 +15,7 @@
 # ===============================================================================
 
 import numpy as np
-from sklearn.decomposition import IncrementalPCA as sklearn_IncrementalPCA
+from sklearn.decomposition import IncrementalPCA as _sklearn_IncrementalPCA
 from sklearn.utils import check_array, gen_batches
 
 from daal4py.sklearn._n_jobs_support import control_n_jobs
@@ -25,11 +25,16 @@ from onedal.decomposition import IncrementalPCA as onedal_IncrementalPCA
 from ..._device_offload import dispatch, wrap_output_data
 from ..._utils import PatchingConditionsChain
 
+if sklearn_check_version("1.6"):
+    from sklearn.utils.validation import validate_data
+else:
+    validate_data = _sklearn_IncrementalPCA._validate_data
+
 
 @control_n_jobs(
     decorated_methods=["fit", "partial_fit", "transform", "_onedal_finalize_fit"]
 )
-class IncrementalPCA(sklearn_IncrementalPCA):
+class IncrementalPCA(_sklearn_IncrementalPCA):
 
     def __init__(self, n_components=None, *, whiten=False, copy=True, batch_size=None):
         super().__init__(
@@ -65,8 +70,8 @@ class IncrementalPCA(sklearn_IncrementalPCA):
 
         if check_input:
             if sklearn_check_version("1.0"):
-                X = self._validate_data(
-                    X, dtype=[np.float64, np.float32], reset=first_pass
+                X = validate_data(
+                    self, X, dtype=[np.float64, np.float32], reset=first_pass
                 )
             else:
                 X = check_array(
@@ -119,7 +124,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
             self._validate_params()
 
         if sklearn_check_version("1.0"):
-            X = self._validate_data(X, dtype=[np.float64, np.float32], copy=self.copy)
+            X = validate_data(self, X, dtype=[np.float64, np.float32], copy=self.copy)
         else:
             X = check_array(
                 X,
@@ -178,7 +183,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
             "partial_fit",
             {
                 "onedal": self.__class__._onedal_partial_fit,
-                "sklearn": sklearn_IncrementalPCA.partial_fit,
+                "sklearn": _sklearn_IncrementalPCA.partial_fit,
             },
             X,
             check_input=check_input,
@@ -191,7 +196,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
             "fit",
             {
                 "onedal": self.__class__._onedal_fit,
-                "sklearn": sklearn_IncrementalPCA.fit,
+                "sklearn": _sklearn_IncrementalPCA.fit,
             },
             X,
         )
@@ -204,7 +209,7 @@ class IncrementalPCA(sklearn_IncrementalPCA):
             "transform",
             {
                 "onedal": self.__class__._onedal_transform,
-                "sklearn": sklearn_IncrementalPCA.transform,
+                "sklearn": _sklearn_IncrementalPCA.transform,
             },
             X,
         )
@@ -216,13 +221,24 @@ class IncrementalPCA(sklearn_IncrementalPCA):
             "fit_transform",
             {
                 "onedal": self.__class__._onedal_fit_transform,
-                "sklearn": sklearn_IncrementalPCA.fit_transform,
+                "sklearn": _sklearn_IncrementalPCA.fit_transform,
             },
             X,
         )
 
-    __doc__ = sklearn_IncrementalPCA.__doc__
-    fit.__doc__ = sklearn_IncrementalPCA.fit.__doc__
-    fit_transform.__doc__ = sklearn_IncrementalPCA.fit_transform.__doc__
-    transform.__doc__ = sklearn_IncrementalPCA.transform.__doc__
-    partial_fit.__doc__ = sklearn_IncrementalPCA.partial_fit.__doc__
+    __doc__ = (
+        _sklearn_IncrementalPCA.__doc__
+        + """
+
+    Note
+    ----
+    Serializing instances of this class will trigger a forced finalization of calculations.
+    Since finalize_fit can't be dispatched without directly provided queue
+    and the dispatching policy can't be serialized, the computation is finalized
+    during serialization call and the policy is not saved in serialized data.
+    """
+    )
+    fit.__doc__ = _sklearn_IncrementalPCA.fit.__doc__
+    fit_transform.__doc__ = _sklearn_IncrementalPCA.fit_transform.__doc__
+    transform.__doc__ = _sklearn_IncrementalPCA.transform.__doc__
+    partial_fit.__doc__ = _sklearn_IncrementalPCA.partial_fit.__doc__

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/preview/decomposition/tests/test_incremental_pca.py

@@ -264,3 +264,73 @@ def test_sklearnex_partial_fit_on_random_data(
     X_df = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
     transformed_data = incpca.transform(X_df)
     check_pca(incpca, dtype, whiten, X, transformed_data)
+
+
+@pytest.mark.parametrize("dataframe,queue", get_dataframes_and_queues())
+@pytest.mark.parametrize("dtype", [np.float32, np.float64])
+def test_sklearnex_incremental_estimatior_pickle(dataframe, queue, dtype):
+    import pickle
+
+    incpca = IncrementalPCA()
+
+    # Check that estimator can be serialized without any data.
+    dump = pickle.dumps(incpca)
+    incpca_loaded = pickle.loads(dump)
+
+    seed = 77
+    gen = np.random.default_rng(seed)
+    X = gen.uniform(low=-0.3, high=+0.7, size=(10, 10))
+    X = X.astype(dtype)
+    X_split = np.array_split(X, 2)
+    X_split_df = _convert_to_dataframe(X_split[0], sycl_queue=queue, target_df=dataframe)
+    incpca.partial_fit(X_split_df)
+    incpca_loaded.partial_fit(X_split_df)
+    dump = pickle.dumps(incpca_loaded)
+    incpca_loaded = pickle.loads(dump)
+    assert incpca.batch_size == incpca_loaded.batch_size
+    assert incpca.n_features_in_ == incpca_loaded.n_features_in_
+    assert incpca.n_samples_seen_ == incpca_loaded.n_samples_seen_
+    if hasattr(incpca, "_parameter_constraints"):
+        assert incpca._parameter_constraints == incpca_loaded._parameter_constraints
+    assert incpca.n_jobs == incpca_loaded.n_jobs
+    X_split_df = _convert_to_dataframe(X_split[1], sycl_queue=queue, target_df=dataframe)
+    incpca.partial_fit(X_split_df)
+    incpca_loaded.partial_fit(X_split_df)
+
+    # Check that estimator can be serialized after partial_fit call.
+    dump = pickle.dumps(incpca)
+    incpca_loaded = pickle.loads(dump)
+
+    assert_allclose(incpca.singular_values_, incpca_loaded.singular_values_, atol=1e-6)
+    assert_allclose(incpca.n_samples_seen_, incpca_loaded.n_samples_seen_, atol=1e-6)
+    assert_allclose(incpca.n_features_in_, incpca_loaded.n_features_in_, atol=1e-6)
+    assert_allclose(incpca.mean_, incpca_loaded.mean_, atol=1e-6)
+    assert_allclose(incpca.var_, incpca_loaded.var_, atol=1e-6)
+    assert_allclose(
+        incpca.explained_variance_, incpca_loaded.explained_variance_, atol=1e-6
+    )
+    assert_allclose(incpca.components_, incpca_loaded.components_, atol=1e-6)
+    assert_allclose(
+        incpca.explained_variance_ratio_,
+        incpca_loaded.explained_variance_ratio_,
+        atol=1e-6,
+    )
+
+    # Check that finalized estimator can be serialized.
+    dump = pickle.dumps(incpca_loaded)
+    incpca_loaded = pickle.loads(dump)
+
+    assert_allclose(incpca.singular_values_, incpca_loaded.singular_values_, atol=1e-6)
+    assert_allclose(incpca.n_samples_seen_, incpca_loaded.n_samples_seen_, atol=1e-6)
+    assert_allclose(incpca.n_features_in_, incpca_loaded.n_features_in_, atol=1e-6)
+    assert_allclose(incpca.mean_, incpca_loaded.mean_, atol=1e-6)
+    assert_allclose(incpca.var_, incpca_loaded.var_, atol=1e-6)
+    assert_allclose(
+        incpca.explained_variance_, incpca_loaded.explained_variance_, atol=1e-6
+    )
+    assert_allclose(incpca.components_, incpca_loaded.components_, atol=1e-6)
+    assert_allclose(
+        incpca.explained_variance_ratio_,
+        incpca_loaded.explained_variance_ratio_,
+        atol=1e-6,
+    )

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_basic_statistics_spmd.py

@@ -18,12 +18,12 @@ import numpy as np
 import pytest
 from numpy.testing import assert_allclose
 
-from onedal.basic_statistics.tests.test_basic_statistics import options_and_tests
+from onedal.basic_statistics.tests.utils import options_and_tests
 from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,
@@ -65,7 +65,7 @@ def test_basic_stats_spmd_gold(dataframe, queue):
     spmd_result = BasicStatistics_SPMD().fit(local_dpt_data)
     batch_result = BasicStatistics_Batch().fit(data)
 
-    for option in (opt[0] for opt in options_and_tests):
+    for option in options_and_tests:
         assert_allclose(getattr(spmd_result, option), getattr(batch_result, option))
 
 
@@ -98,7 +98,7 @@ def test_basic_stats_spmd_synthetic(n_samples, n_features, dataframe, queue, dty
     batch_result = BasicStatistics_Batch().fit(data)
 
     tol = 1e-5 if dtype == np.float32 else 1e-7
-    for option in (opt[0] for opt in options_and_tests):
+    for option in options_and_tests:
         assert_allclose(
             getattr(spmd_result, option),
             getattr(batch_result, option),

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/basic_statistics/tests/test_incremental_basic_statistics_spmd.py

@@ -18,12 +18,12 @@ import numpy as np
 import pytest
 from numpy.testing import assert_allclose
 
-from onedal.basic_statistics.tests.test_basic_statistics import options_and_tests
+from onedal.basic_statistics.tests.utils import options_and_tests
 from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,
@@ -87,7 +87,7 @@ def test_incremental_basic_statistics_fit_spmd_gold(dataframe, queue, weighted,
         dpt_data, sample_weight=dpt_weights if weighted else None
     )
 
-    for option, _, _ in options_and_tests:
+    for option in options_and_tests:
         assert_allclose(
             getattr(incbs_spmd, option),
             getattr(incbs, option),
@@ -160,7 +160,7 @@ def test_incremental_basic_statistics_partial_fit_spmd_gold(
 
     incbs.fit(dpt_data, sample_weight=dpt_weights if weighted else None)
 
-    for option, _, _ in options_and_tests:
+    for option in options_and_tests:
         assert_allclose(
             getattr(incbs_spmd, option),
             getattr(incbs, option),
@@ -178,11 +178,11 @@ def test_incremental_basic_statistics_partial_fit_spmd_gold(
 )
 @pytest.mark.parametrize("num_blocks", [1, 2])
 @pytest.mark.parametrize("weighted", [True, False])
-@pytest.mark.parametrize("result_option", options_and_tests)
+@pytest.mark.parametrize("option", options_and_tests.keys())
 @pytest.mark.parametrize("dtype", [np.float32, np.float64])
 @pytest.mark.mpi
 def test_incremental_basic_statistics_single_option_partial_fit_spmd_gold(
-    dataframe, queue, num_blocks, weighted, result_option, dtype
+    dataframe, queue, num_blocks, weighted, option, dtype
 ):
     # Import spmd and batch algo
     from sklearnex.basic_statistics import IncrementalBasicStatistics
@@ -217,7 +217,6 @@ def test_incremental_basic_statistics_single_option_partial_fit_spmd_gold(
         local_weights = _get_local_tensor(weights)
         split_local_weights = np.array_split(local_weights, num_blocks)
 
-    option, _, _ = result_option
     incbs_spmd = IncrementalBasicStatistics_SPMD(result_options=option)
     incbs = IncrementalBasicStatistics(result_options=option)
 
@@ -298,7 +297,7 @@ def test_incremental_basic_statistics_partial_fit_spmd_synthetic(
         )
         incbs.partial_fit(dpt_data, sample_weight=dpt_weights if weighted else None)
 
-    for option, _, _ in options_and_tests:
+    for option in options_and_tests:
         assert_allclose(
             getattr(incbs_spmd, option),
             getattr(incbs, option),

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_dbscan_spmd.py

@@ -21,7 +21,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_clustering_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/cluster/tests/test_kmeans_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _assert_kmeans_labels_allclose,
     _assert_unordered_allclose,
     _generate_clustering_data,
@@ -59,9 +59,10 @@ def test_kmeans_spmd_gold(dataframe, queue):
             [2, 2],
             [1, 3],
             [2, 2],
-        ]
+        ],
+        dtype=np.float64,
     )
-    X_test = np.array([[0, 0], [12, 3], [2, 2], [7, 8]])
+    X_test = np.array([[0, 0], [12, 3], [2, 2], [7, 8]], dtype=np.float64)
 
     local_dpt_X_train = _convert_to_dataframe(
         _get_local_tensor(X_train), sycl_queue=queue, target_df=dataframe

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_covariance_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/covariance/tests/test_incremental_covariance_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py

@@ -23,7 +23,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/decomposition/tests/test_pca_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_statistic_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/ensemble/tests/test_forest_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_classification_data,
     _generate_regression_data,
     _get_local_tensor,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_incremental_linear_spmd.py

@@ -23,7 +23,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_regression_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,
@@ -86,20 +86,21 @@ def test_incremental_linear_regression_fit_spmd_gold(
     inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
 
     if macro_block is not None:
-        hparams = inclin.get_hyperparameters("fit")
+        hparams = IncrementalLinearRegression.get_hyperparameters("fit")
         hparams.cpu_macro_block = macro_block
         hparams.gpu_macro_block = macro_block
 
-        hparams_spmd = inclin_spmd.get_hyperparameters("fit")
+        hparams_spmd = IncrementalLinearRegression_SPMD.get_hyperparameters("fit")
         hparams_spmd.cpu_macro_block = macro_block
         hparams_spmd.gpu_macro_block = macro_block
 
     inclin_spmd.fit(local_dpt_X, local_dpt_y)
     inclin.fit(dpt_X, dpt_y)
 
-    assert_allclose(inclin.coef_, inclin_spmd.coef_)
+    rtol = 1e-5 if (dtype == np.float32) else 1e-7
+    assert_allclose(inclin.coef_, inclin_spmd.coef_, rtol=rtol)
     if fit_intercept:
-        assert_allclose(inclin.intercept_, inclin_spmd.intercept_)
+        assert_allclose(inclin.intercept_, inclin_spmd.intercept_, rtol=rtol)
 
 
 @pytest.mark.skipif(
@@ -159,11 +160,11 @@ def test_incremental_linear_regression_partial_fit_spmd_gold(
     inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
 
     if macro_block is not None:
-        hparams = inclin.get_hyperparameters("fit")
+        hparams = IncrementalLinearRegression.get_hyperparameters("fit")
         hparams.cpu_macro_block = macro_block
         hparams.gpu_macro_block = macro_block
 
-        hparams_spmd = inclin_spmd.get_hyperparameters("fit")
+        hparams_spmd = IncrementalLinearRegression_SPMD.get_hyperparameters("fit")
         hparams_spmd.cpu_macro_block = macro_block
         hparams_spmd.gpu_macro_block = macro_block
 
@@ -178,9 +179,10 @@ def test_incremental_linear_regression_partial_fit_spmd_gold(
 
     inclin.fit(dpt_X, dpt_y)
 
-    assert_allclose(inclin.coef_, inclin_spmd.coef_)
+    rtol = 1e-5 if (dtype == np.float32) else 1e-7
+    assert_allclose(inclin.coef_, inclin_spmd.coef_, rtol=rtol)
     if fit_intercept:
-        assert_allclose(inclin.intercept_, inclin_spmd.intercept_)
+        assert_allclose(inclin.intercept_, inclin_spmd.intercept_, rtol=rtol)
 
 
 @pytest.mark.skipif(
@@ -206,7 +208,7 @@ def test_incremental_linear_regression_fit_spmd_random(
         IncrementalLinearRegression as IncrementalLinearRegression_SPMD,
     )
 
-    tol = 2e-4 if dtype == np.float32 else 1e-7
+    tol = 5e-3 if dtype == np.float32 else 1e-7
 
     # Generate random data and process into dpt
     X_train, X_test, y_train, _ = _generate_regression_data(
@@ -225,11 +227,11 @@ def test_incremental_linear_regression_fit_spmd_random(
     inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
 
     if macro_block is not None:
-        hparams = inclin.get_hyperparameters("fit")
+        hparams = IncrementalLinearRegression.get_hyperparameters("fit")
         hparams.cpu_macro_block = macro_block
         hparams.gpu_macro_block = macro_block
 
-        hparams_spmd = inclin_spmd.get_hyperparameters("fit")
+        hparams_spmd = IncrementalLinearRegression_SPMD.get_hyperparameters("fit")
         hparams_spmd.cpu_macro_block = macro_block
         hparams_spmd.gpu_macro_block = macro_block
 
@@ -277,7 +279,7 @@ def test_incremental_linear_regression_partial_fit_spmd_random(
         IncrementalLinearRegression as IncrementalLinearRegression_SPMD,
     )
 
-    tol = 3e-4 if dtype == np.float32 else 1e-7
+    tol = 5e-3 if dtype == np.float32 else 1e-7
 
     # Generate random data and process into dpt
     X_train, X_test, y_train, _ = _generate_regression_data(
@@ -298,11 +300,11 @@ def test_incremental_linear_regression_partial_fit_spmd_random(
     inclin = IncrementalLinearRegression(fit_intercept=fit_intercept)
 
     if macro_block is not None:
-        hparams = inclin.get_hyperparameters("fit")
+        hparams = IncrementalLinearRegression.get_hyperparameters("fit")
         hparams.cpu_macro_block = macro_block
         hparams.gpu_macro_block = macro_block
 
-        hparams_spmd = inclin_spmd.get_hyperparameters("fit")
+        hparams_spmd = IncrementalLinearRegression_SPMD.get_hyperparameters("fit")
         hparams_spmd.cpu_macro_block = macro_block
         hparams_spmd.gpu_macro_block = macro_block
 

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_linear_regression_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_regression_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py

@@ -23,7 +23,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _generate_classification_data,
     _get_local_tensor,
     _mpi_libs_and_gpu_available,
@@ -116,10 +116,6 @@ def test_logistic_spmd_gold(dataframe, queue):
 @pytest.mark.parametrize("dtype", [np.float32, np.float64])
 @pytest.mark.mpi
 def test_logistic_spmd_synthetic(n_samples, n_features, C, tol, dataframe, queue, dtype):
-    if dtype == np.float64:
-        pytest.skip(
-            "Sporadic failures on coef_ check. Test disabled while fix in progress"
-        )
     # TODO: Resolve numerical issues when n_rows_rank < n_cols
     if n_samples <= n_features:
         pytest.skip("Numerical issues when rank rows < columns")

{scikit_learn_intelex-2025.0.1.data → scikit_learn_intelex-2025.2.0.data}/data/Lib/site-packages/sklearnex/spmd/neighbors/tests/test_neighbors_spmd.py

@@ -22,7 +22,7 @@ from onedal.tests.utils._dataframes_support import (
     _convert_to_dataframe,
     get_dataframes_and_queues,
 )
-from sklearnex.tests._utils_spmd import (
+from sklearnex.tests.utils.spmd import (
     _assert_unordered_allclose,
     _generate_classification_data,
     _generate_regression_data,