From 099f362fd3feaf9d208cc62073dfcd18f3f5f058 Mon Sep 17 00:00:00 2001 From: Martin Stancsics Date: Tue, 23 Jan 2024 04:11:38 +0100 Subject: [PATCH] Make pre-commit happy --- tests/glm/test_glm.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/glm/test_glm.py b/tests/glm/test_glm.py index 0829722e..e9d2bb3a 100644 --- a/tests/glm/test_glm.py +++ b/tests/glm/test_glm.py @@ -3210,7 +3210,9 @@ def test_cat_missing(cat_missing_method, unseen_missing, formula): fit_intercept=False, ) if cat_missing_method == "fail" and not unseen_missing: - with pytest.raises(ValueError, match="Categorical data can't have missing values"): + with pytest.raises( + ValueError, match="Categorical data can't have missing values" + ): model.fit(X, y) else: model.fit(X, y) @@ -3224,7 +3226,9 @@ def test_cat_missing(cat_missing_method, unseen_missing, formula): assert len(model.coef_) == len(feature_names) if cat_missing_method == "fail" and unseen_missing: - with pytest.raises(ValueError, match="Categorical data can't have missing values"): + with pytest.raises( + ValueError, match="Categorical data can't have missing values" + ): model.predict(X_unseen) elif cat_missing_method == "convert" and unseen_missing: with pytest.raises(ValueError, match="contains unseen categories"):