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"):