Skip to content

Commit 4346b23

Browse files
committed
Merge branch 'master' of github.com:DoubleML/doubleml-for-py into m-pliv-iv-type
2 parents 8875602 + 5f5e610 commit 4346b23

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doubleml/datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def fetch_bonus(return_type='DoubleMLData', polynomial_features=False):
104104
if polynomial_features:
105105
poly = PolynomialFeatures(2, include_bias=False)
106106
data_transf = poly.fit_transform(data[x_cols])
107-
x_cols = poly.get_feature_names(x_cols)
107+
x_cols = list(poly.get_feature_names_out(x_cols))
108108

109109
data_transf = pd.DataFrame(data_transf, columns=x_cols)
110110
data = pd.concat((data[[y_col] + d_cols], data_transf),

0 commit comments

Comments
 (0)