DoubleML · SvenKlaassen · Nov 21, 2025 · Sep 8, 2025 · Sep 23, 2025 · Sep 23, 2025
diff --git a/.devcontainer/Dockerfile.dev b/.devcontainer/Dockerfile.dev
@@ -62,7 +62,7 @@ COPY --chown=$USERNAME:$USERNAME requirements.txt /tmp/requirements.txt
 RUN python -m venv /home/$USERNAME/.venv && \
     /home/$USERNAME/.venv/bin/python -m pip install --upgrade pip && \
     /home/$USERNAME/.venv/bin/pip install --no-cache-dir -r /tmp/requirements.txt && \
-    /home/$USERNAME/.venv/bin/pip install --no-cache-dir git+https://github.com/DoubleML/doubleml-for-py.git@main#egg=DoubleML[rdd]
+    /home/$USERNAME/.venv/bin/pip install --no-cache-dir "DoubleML[rdd] @ git+https://github.com/DoubleML/doubleml-for-py.git@main"
 
 # Set the virtual environment as the default Python environment
 ENV PATH="/home/$USERNAME/.venv/bin:$PATH"

diff --git a/doc/api/data_class.rst b/doc/api/data_class.rst
@@ -12,3 +12,6 @@ DoubleML Data Class
     DoubleMLData
     DoubleMLClusterData
     DoubleMLPanelData
+    DoubleMLSSMData
+    DoubleMLRDDData
+    DoubleMLDIDData
diff --git a/doc/api/datasets.rst b/doc/api/datasets.rst
@@ -6,13 +6,13 @@ Datasets
 Dataset Loaders
 ~~~~~~~~~~~~~~~
 
-.. currentmodule:: doubleml
+.. currentmodule:: doubleml.datasets
 
 .. autosummary::
    :toctree: generated/
 
-   datasets.fetch_401K
-   datasets.fetch_bonus
+   fetch_401K
+   fetch_bonus
 
 Dataset Generators
 ~~~~~~~~~~~~~~~~~~
@@ -22,18 +22,22 @@ Dataset Generators
 .. autosummary::
    :toctree: generated/
 
-   datasets.make_plr_CCDDHNR2018
-   datasets.make_pliv_CHS2015
-   datasets.make_irm_data
-   datasets.make_iivm_data
-   datasets.make_plr_turrell2018
-   datasets.make_pliv_multiway_cluster_CKMS2021
-
-   datasets.make_ssm_data
-   datasets.make_confounded_plr_data
-   datasets.make_confounded_irm_data
-   datasets.make_heterogeneous_data
-   datasets.make_irm_data_discrete_treatments
+   irm.datasets.make_irm_data
+   irm.datasets.make_iivm_data
+   irm.datasets.make_heterogeneous_data
+   irm.datasets.make_irm_data_discrete_treatments
+   irm.datasets.make_confounded_irm_data
+   irm.datasets.make_ssm_data
+
+   plm.datasets.make_plr_CCDDHNR2018
+   plm.datasets.make_plr_turrell2018
+   plm.datasets.make_lplr_LZZ2020
+   plm.datasets.make_pliv_CHS2015
+   plm.datasets.make_pliv_multiway_cluster_CKMS2021
+   plm.datasets.make_confounded_plr_data
+
    did.datasets.make_did_SZ2020
    did.datasets.make_did_CS2021
-   rdd.datasets.make_simple_rdd_data
+   did.datasets.make_did_cs_CS2021
+
+   rdd.datasets.make_simple_rdd_data
diff --git a/doc/api/dml_models.rst b/doc/api/dml_models.rst
@@ -16,6 +16,7 @@ doubleml.plm
     :template: class.rst
 
     DoubleMLPLR
+    DoubleMLLPLR
     DoubleMLPLIV
 
 

diff --git a/doc/api/utility.rst b/doc/api/utility.rst
@@ -18,6 +18,8 @@ Utility Classes
     utils.DoubleMLPolicyTree
     utils.GlobalRegressor
     utils.GlobalClassifier
+    utils.PSProcessorConfig
+    utils.PSProcessor
 
 Utility Functions
 ~~~~~~~~~~~~~~~~~

diff --git a/doc/conf.py b/doc/conf.py
@@ -94,8 +94,8 @@
     "navbar_align": "left",
     "logo": {
         "text": project,
-        "image_light": "img/logo.png",
-        "image_dark": "img/logo_dark.png",
+        "image_light": "../img/logo.png",
+        "image_dark": "../img/logo_dark.png",
     },
     "header_links_before_dropdown": 6,
     "use_edit_page_button": True,
@@ -133,6 +133,7 @@
 html_favicon = "../img/favicon.ico"
 
 html_sidebars = {
+
     "**": [
         "logo.html",
         "search-field.html",
@@ -273,6 +274,8 @@
     "https://doi.org/10.1097%2FEDE.0b013e3181f74493",
     # Valid DOI; Causes 403 Client Error: Forbidden for url:...
     "https://doi.org/10.3982/ECTA15732",
+    # Valid DOI; Causes 403 Client Error: Forbidden for url:...
+    "https://doi.org/10.1093/ectj/utab019"
 ]
 
 # To execute R code via jupyter-execute one needs to install the R kernel for jupyter

diff --git a/doc/examples/did/py_did.ipynb b/doc/examples/did/py_did.ipynb
diff --git a/doc/examples/did/py_did_pretest.ipynb b/doc/examples/did/py_did_pretest.ipynb
diff --git a/doc/examples/did/py_panel.ipynb b/doc/examples/did/py_panel.ipynb
@@ -367,6 +367,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "np.random.seed(42)\n",
     "dml_obj = DoubleMLDIDMulti(dml_data, **default_args)\n",
     "dml_obj.fit()\n",
     "print(dml_obj)"

diff --git a/doc/examples/did/py_panel_simple.ipynb b/doc/examples/did/py_panel_simple.ipynb
@@ -250,7 +250,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Event Study Aggregation\n",
+    "### Event Study Aggregation\n",
     "\n",
     "Finally, `aggregation=\"eventstudy\"` aggregates $\\widehat{ATT}(\\mathrm{g},t_\\text{pre},t_\\text{eval})$ based on exposure time $e = t_\\text{eval} - \\mathrm{g}$ (respecting group size)."
    ]

diff --git a/doc/examples/did/py_rep_cs.ipynb b/doc/examples/did/py_rep_cs.ipynb
@@ -50,7 +50,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "n_obs = 5000\n",
+    "n_obs = 10000\n",
     "n_periods = 6\n",
     "\n",
     "df = make_did_cs_CS2021(n_obs, dgp_type=4, include_never_treated=True, n_periods=n_periods, n_pre_treat_periods=3,\n",
@@ -951,7 +951,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "dml_dev",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -965,7 +965,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.8"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

diff --git a/doc/examples/index.rst b/doc/examples/index.rst
@@ -22,10 +22,11 @@ General Examples
     py_double_ml_sensitivity.ipynb
     py_double_ml_apo.ipynb
     py_double_ml_irm_vs_apo.ipynb
+    py_double_ml_lplr.ipynb
+    py_double_ml_ssm.ipynb
     py_double_ml_learner.ipynb
     py_double_ml_firststage.ipynb
     py_double_ml_multiway_cluster.ipynb
-    py_double_ml_ssm.ipynb
     py_double_ml_sensitivity_booking.ipynb
     learners/py_tabpfn.ipynb
     py_double_ml_basic_iv.ipynb

diff --git a/doc/examples/learners/py_tabpfn.ipynb b/doc/examples/learners/py_tabpfn.ipynb
diff --git a/doc/examples/py_double_ml_apo.ipynb b/doc/examples/py_double_ml_apo.ipynb
diff --git a/doc/examples/py_double_ml_basic_iv.ipynb b/doc/examples/py_double_ml_basic_iv.ipynb
@@ -186,7 +186,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -200,7 +200,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.5"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

diff --git a/doc/examples/py_double_ml_basics.ipynb b/doc/examples/py_double_ml_basics.ipynb
diff --git a/doc/examples/py_double_ml_cate.ipynb b/doc/examples/py_double_ml_cate.ipynb
@@ -22,7 +22,7 @@
     "import pandas as pd\n",
     "import doubleml as dml\n",
     "\n",
-    "from doubleml.datasets import make_heterogeneous_data"
+    "from doubleml.irm.datasets import make_heterogeneous_data"
    ]
   },
   {
@@ -395,7 +395,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3.10.6 64-bit",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -410,11 +410,6 @@
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
    "version": "3.12.3"
-  },
-  "vscode": {
-   "interpreter": {
-    "hash": "ac5e9af40c2048901fb5e070f7bbe2ca12417b0669992742e66f016e0e17b88e"
-   }
   }
  },
  "nbformat": 4,

diff --git a/doc/examples/py_double_ml_cate_plr.ipynb b/doc/examples/py_double_ml_cate_plr.ipynb
@@ -19,7 +19,7 @@
     "import pandas as pd\n",
     "import doubleml as dml\n",
     "\n",
-    "from doubleml.datasets import make_heterogeneous_data"
+    "from doubleml.irm.datasets import make_heterogeneous_data"
    ]
   },
   {
@@ -388,7 +388,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "dml_dev",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -402,7 +402,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.2"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

diff --git a/doc/examples/py_double_ml_cvar.ipynb b/doc/examples/py_double_ml_cvar.ipynb
diff --git a/doc/examples/py_double_ml_firststage.ipynb b/doc/examples/py_double_ml_firststage.ipynb
@@ -216,7 +216,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "doubleml",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -230,7 +230,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.5"
+   "version": "3.12.3"
   },
   "orig_nbformat": 4
  },