fix correct predict method assignment for binary outcome in PLR

SvenKlaassen · SvenKlaassen · commit ec91b159f834 · 2025-10-20T10:12:35.000+02:00
diff --git a/doubleml/plm/plr.py b/doubleml/plm/plr.py
@@ -121,7 +121,7 @@ def __init__(
 
         if ml_l_is_classifier:
             if obj_dml_data.binary_outcome:
-                self._predict_method = {"ml_g": "predict_proba"}
+                self._predict_method = {"ml_l": "predict_proba"}
                 warnings.warn(
                     f"The ml_l learner {str(ml_l)} was identified as classifier. Fitting an additive probability model."
                 )

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@ def __init__(`
`121`	`121`
`122`	`122`	`if ml_l_is_classifier:`
`123`	`123`	`if obj_dml_data.binary_outcome:`
`124`		`- self._predict_method = {"ml_g": "predict_proba"}`
	`124`	`+ self._predict_method = {"ml_l": "predict_proba"}`
`125`	`125`	`warnings.warn(`
`126`	`126`	`f"The ml_l learner {str(ml_l)} was identified as classifier. Fitting an additive probability model."`
`127`	`127`	`)`