switch to a different temporary fix for #45 / scikit-learn/scikit-learn#21997

Author: MalteKurz

doc/examples/py_double_ml_pension.ipynb

@@ -91,6 +91,23 @@
     "data = fetch_401K(return_type='DataFrame')"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "853ba4be",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Temporary fix for https://github.com/DoubleML/doubleml-docs/issues/45 / https://github.com/scikit-learn/scikit-learn/issues/21997\n",
+    "# Can be removed when scikit-learn version 1.2.0 is released\n",
+    "dtypes = data.dtypes\n",
+    "dtypes['nifa'] = 'float64'\n",
+    "dtypes['net_tfa'] = 'float64'\n",
+    "dtypes['tw'] = 'float64'\n",
+    "dtypes['inc'] = 'float64'\n",
+    "data = data.astype(dtypes)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,

requirements.txt

@@ -1,4 +1,3 @@
-scikit-learn<1.0.0
 DoubleML
 
 # test