10000 TST adapt tol for ridge tests to pass on all random seeds (#23017) · thomasjpfan/scikit-learn@566cc46 · GitHub
[go: up one dir, main page]

Skip to content

Commit 566cc46

Browse files
authored
TST adapt tol for ridge tests to pass on all random seeds (scikit-learn#23017)
1 parent d7a0596 commit 566cc46

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sklearn/linear_model/tests/test_ridge.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ def test_ridge_regression_vstacked_X(
256256
alpha=2 * alpha,
257257
fit_intercept=fit_intercept,
258258
solver=solver,
259-
tol=1e-11,
259+
tol=1e-15 if solver in ("sag", "saga") else 1e-10,
260260
random_state=global_random_seed,
261261
)
262262
X = X[:, :-1] # remove intercept
@@ -1663,7 +1663,7 @@ def test_ridge_fit_intercept_sparse(solver, with_sample_weight, global_random_se
16631663
sparse_ridge.fit(sp.csr_matrix(X), y, sample_weight=sample_weight)
16641664

16651665
assert_allclose(dense_ridge.intercept_, sparse_ridge.intercept_)
1666-
assert_allclose(dense_ridge.coef_, sparse_ridge.coef_)
1666+
assert_allclose(dense_ridge.coef_, sparse_ridge.coef_, rtol=5e-7)
16671667

16681668

16691669
@pytest.mark.parametrize("solver", ["saga", "svd", "cholesky"])

0 commit comments

Comments
 (0)
0