fix notebook warnings

Author: SvenKlaassen

doc/examples/py_double_ml_did.ipynb

@@ -181,8 +181,8 @@
     "\n",
     "    ATTE_estimates[i_rep] = dml_did.coef\n",
     "    confint = dml_did.confint(level=0.95)\n",
-    "    coverage [i_rep] = (confint['2.5 %'] <= ATTE) & (ATTE <= confint['97.5 %'])\n",
-    "    ci_length[i_rep] = confint['97.5 %'] - confint['2.5 %']"
+    "    coverage [i_rep] = (confint['2.5 %'].iloc[0] <= ATTE) & (ATTE <= confint['97.5 %'].iloc[0])\n",
+    "    ci_length[i_rep] = confint['97.5 %'].iloc[0] - confint['2.5 %'].iloc[0]"
    ]
   },
   {
@@ -384,8 +384,8 @@
     "\n",
     "    ATTE_estimates[i_rep] = dml_did.coef\n",
     "    confint = dml_did.confint(level=0.95)\n",
-    "    coverage [i_rep] = (confint['2.5 %'] <= ATTE) & (ATTE <= confint['97.5 %'])\n",
-    "    ci_length[i_rep] = confint['97.5 %'] - confint['2.5 %']"
+    "    coverage [i_rep] = (confint['2.5 %'].iloc[0] <= ATTE) & (ATTE <= confint['97.5 %'].iloc[0])\n",
+    "    ci_length[i_rep] = confint['97.5 %'].iloc[0] - confint['2.5 %'].iloc[0]"
    ]
   },
   {

doc/examples/py_double_ml_learner.ipynb

@@ -124,7 +124,7 @@
     "        rmses_ml_g1[i_rep, i_learners] = dml_irm.rmses['ml_g1']\n",
     "\n",
     "        confint = dml_irm.confint()\n",
-    "        coverage[i_rep, i_learners] = (confint['2.5 %'] <= theta) & (confint['97.5 %'] >= theta)\n",
+    "        coverage[i_rep, i_learners] = (confint['2.5 %'].iloc[0] <= theta) & (confint['97.5 %'].iloc[0]s >= theta)\n",
     "\n",
     "print(f'\\nCoverage: {coverage.mean(0)}')"
    ]
@@ -446,7 +446,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.8"
+   "version": "3.10.10"
   },
   "orig_nbformat": 4
  },