From 7f394b340a35c56c217128929984c9b2d7f0e76c Mon Sep 17 00:00:00 2001 From: TomeHirata Date: Sun, 21 Sep 2025 18:24:03 +0900 Subject: [PATCH 1/2] fix empirical LPTE issue --- dte_adj/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dte_adj/util.py b/dte_adj/util.py index 1ca1e3a..39026f8 100644 --- a/dte_adj/util.py +++ b/dte_adj/util.py @@ -185,7 +185,7 @@ def _compute_local_treatment_effects_core( # Compute outcome indicators (different for LDTE vs LPTE) if use_intervals: - bi = (Y[i] < locations) * 1 + bi = (Y[i] <= locations) * 1 bi = bi[1:] - bi[:-1] # Convert to interval probabilities else: bi = Y[i] <= locations From 917468e93b1d5ab4e9f4fef6ce11eae6e660851e Mon Sep 17 00:00:00 2001 From: TomeHirata Date: Sun, 21 Sep 2025 18:25:58 +0900 Subject: [PATCH 2/2] nit --- dte_adj/util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dte_adj/util.py b/dte_adj/util.py index 39026f8..c0265b3 100644 --- a/dte_adj/util.py +++ b/dte_adj/util.py @@ -188,7 +188,7 @@ def _compute_local_treatment_effects_core( bi = (Y[i] <= locations) * 1 bi = bi[1:] - bi[:-1] # Convert to interval probabilities else: - bi = Y[i] <= locations + bi = (Y[i] <= locations) * 1 xi_t[i] = ((1 - 1 / w_s) * y_t_mu[i] - y_c_mu[i] + bi / w_s) - beta * ( (1 - 1 / w_s) * d_t_eta[i] - d_c_eta[i] + D[i] / w_s