Skip to content

Commit 80d6c9c

Browse files
committed
error message was changed in sklearn==1.1.0
1 parent f5d5985 commit 80d6c9c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doubleml/tests/test_irm_with_missings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,6 @@ def test_irm_exception_with_missings(generate_data_irm_w_missings, learner_sklea
144144
dml_irm_obj = dml.DoubleMLIRM(obj_dml_data,
145145
ml_g, ml_m)
146146

147-
msg = r"Input contains NaN, infinity or a value too large for dtype\('float64'\)."
147+
msg = r"Input X contains NaN.\nLinearRegression does not accept missing values encoded as NaN natively."
148148
with pytest.raises(ValueError, match=msg):
149149
dml_irm_obj.fit()

0 commit comments

Comments
 (0)