diff --git a/tests/glm/test_glm.py b/tests/glm/test_glm.py index 013890d7..469f464e 100644 --- a/tests/glm/test_glm.py +++ b/tests/glm/test_glm.py @@ -389,14 +389,14 @@ def test_P1_P2_expansion_with_categoricals_missings(): mdl2.fit(X, y) np.testing.assert_allclose(mdl1.coef_, mdl2.coef_) - mdl2 = GeneralizedLinearRegressor( + mdl3 = GeneralizedLinearRegressor( l1_ratio=0.01, P1=[1, 2], P2=sparse.diags([2, 1, 1, 1, 1, 1]), cat_missing_method="convert", ) - mdl2.fit(X, y) - np.testing.assert_allclose(mdl1.coef_, mdl2.coef_) + mdl3.fit(X, y) + np.testing.assert_allclose(mdl1.coef_, mdl3.coef_) @pytest.mark.parametrize(