Skip to content

Commit 0378ee6

Browse files
committed
Fix: Remove deprecated 'multi_class' parameter from LogisticRegression instantiation
1 parent 4716039 commit 0378ee6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

doubleml/tests/test_optuna_multi_wrappers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def _build_apos_object():
2323
dml_data = dml.DoubleMLData(df, "y", "d")
2424

2525
ml_g = LinearRegression()
26-
ml_m = LogisticRegression(max_iter=200, multi_class="auto")
26+
ml_m = LogisticRegression(max_iter=200)
2727

2828
return dml.DoubleMLAPOS(
2929
dml_data,
@@ -38,7 +38,7 @@ def _build_apos_object():
3838
def _build_qte_object():
3939
np.random.seed(3141)
4040
dml_data = make_irm_data(n_obs=80, dim_x=5)
41-
ml = LogisticRegression(max_iter=200, multi_class="auto")
41+
ml = LogisticRegression(max_iter=200)
4242

4343
return dml.DoubleMLQTE(
4444
dml_data,
@@ -57,7 +57,7 @@ def _build_did_multi_object():
5757
dml_panel = DoubleMLPanelData(df, y_col="y", d_cols="d", t_col="t", id_col="id", x_cols=x_cols)
5858

5959
ml_g = LinearRegression()
60-
ml_m = LogisticRegression(max_iter=200, multi_class="auto")
60+
ml_m = LogisticRegression(max_iter=200)
6161

6262
return dml.did.DoubleMLDIDMulti(
6363
obj_dml_data=dml_panel,

0 commit comments

Comments
 (0)