From ae3c3a517f016b6c920512f0b22c125cafa16452 Mon Sep 17 00:00:00 2001 From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com> Date: Wed, 28 Aug 2024 15:43:42 +0100 Subject: [PATCH] Logistic Regression CI fix (#2013) (#2017) * increase tolerance * increase tolerance * increase tolerance (cherry picked from commit d7a2f92e7e620de5e180f3e1a788176e1d61e3f0) Co-authored-by: msa <111298646+md-shafiul-alam@users.noreply.github.com> --- sklearnex/linear_model/tests/test_logreg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearnex/linear_model/tests/test_logreg.py b/sklearnex/linear_model/tests/test_logreg.py index 4b47b2c517..8ee05cd8b5 100755 --- a/sklearnex/linear_model/tests/test_logreg.py +++ b/sklearnex/linear_model/tests/test_logreg.py @@ -129,5 +129,5 @@ def test_csr(queue, dtype, dims): assert_allclose(pred, pred_sp) assert_allclose(prob, prob_sp) - assert_allclose(model.coef_, model_sp.coef_) - assert_allclose(model.intercept_, model_sp.intercept_) + assert_allclose(model.coef_, model_sp.coef_, rtol=1e-4) + assert_allclose(model.intercept_, model_sp.intercept_, rtol=1e-4)