Merge branch 'main' into s-notebook-CATE

Author: SvenKlaassen

.gitignore

@@ -1 +1,2 @@
-*idea
+*.idea
+

doc/examples/py_double_ml_multiway_cluster.ipynb

@@ -607,7 +607,7 @@
    "source": [
     "poly = PolynomialFeatures(degree=3, include_bias=False)\n",
     "data_transf = poly.fit_transform(blp_data[x_cols])\n",
-    "x_cols_poly = poly.get_feature_names(x_cols)\n",
+    "x_cols_poly = poly.get_feature_names_out(x_cols)\n",
     "data_transf = pd.DataFrame(data_transf, columns=x_cols_poly)\n",
     "data_transf.index = blp_data.index"
    ]

doc/examples/py_double_ml_pension.ipynb

@@ -343,12 +343,11 @@
     "for key, degree in poly_dict.items():\n",
     "    poly = PolynomialFeatures(degree, include_bias=False)\n",
     "    data_transf = poly.fit_transform(data[[key]])\n",
-    "    x_cols = poly.get_feature_names([key])\n",
+    "    x_cols = poly.get_feature_names_out([key])\n",
     "    data_transf = pd.DataFrame(data_transf, columns=x_cols)\n",
     "    \n",
     "    features = pd.concat((features, data_transf),\n",
     "                          axis=1, sort=False)\n",
-    "\n",
     "model_data = pd.concat((data.copy()[['net_tfa', 'e401']], features.copy()),\n",
     "                        axis=1, sort=False)\n",
     "\n",