From cf29d09aa7db6e8556e061bb30402fdcc93fed47 Mon Sep 17 00:00:00 2001 From: Anatoly Volkov <117643568+avolkov-intel@users.noreply.github.com> Date: Tue, 24 Sep 2024 13:38:32 -0700 Subject: [PATCH] TEST: Enable spmd logreg test (#2065) --- .../spmd/linear_model/tests/test_logistic_regression_spmd.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py b/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py index bfe8203bab..8113f55f5a 100644 --- a/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py +++ b/sklearnex/spmd/linear_model/tests/test_logistic_regression_spmd.py @@ -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")