From 2756dec47c016aeb1d8c92e8000530c9e6d947b5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?=
 <francesc.marti.escofet@quantco.com>
Date: Wed, 10 Jul 2024 14:13:51 +0200
Subject: [PATCH] Add docstring

---
 metalearners/drlearner.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/metalearners/drlearner.py b/metalearners/drlearner.py
index cea00203..9d3379b4 100644
--- a/metalearners/drlearner.py
+++ b/metalearners/drlearner.py
@@ -59,6 +59,9 @@ class DRLearner(_ConditionalAverageOutcomeMetaLearner):
 
         * ``"treatment_model"`` which estimates :math:`\mathbb{E}[Y(k) - Y(0) | X]`
 
+    If ``adaptive_clipping`` is set to ``True``, then the pseudo outcomes are computed using
+    adaptive propensity clipping described in section 4.1, equation *DR-Switch* of
+    `Mahajan et al. (2024) <https://arxiv.org/pdf/2211.01939>`_.
     """
 
     @classmethod