add dgp and update notebook

Author: SvenKlaassen

doc/api/api.rst

@@ -63,6 +63,7 @@ Dataset generators
    datasets.make_iivm_data
    datasets.make_plr_turrell2018
    datasets.make_pliv_multiway_cluster_CKMS2021
+   datasets.make_did_SZ2020
 
 Score mixin classes for double machine learning models
 ------------------------------------------------------

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

doc/shared/models/did.rst

@@ -15,7 +15,7 @@ Target parameter of interest is the average treatment effect on the treated (ATT
 
     \theta_0 = \mathbb{E}[Y_{i1}(1)- Y_{i1}(0)|D_i=1].
 
-The corresponding identifying assumptions are the 
+The corresponding identifying assumptions are
 
  - **(Cond.) Parallel Trends:** :math:`\mathbb{E}[Y_{i1}(0) - Y_{i0}(0)|X_i, D_i=1] = \mathbb{E}[Y_{i1}(0) - Y_{i0}(0)|X_i, D_i=0]\quad a.s.`
  - **Overlap:** :math:`\exists\epsilon > 0`: :math:`P(D_i=1) > \epsilon` and :math:`P(D_i=1|X_i) \le 1-\epsilon\quad a.s.`