Skip to content

Commit 2fc68cc

Browse files
authored
Fix empirical LPTE issue (#73)
* fix empirical LPTE issue * nit
1 parent 1208350 commit 2fc68cc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

dte_adj/util.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -185,10 +185,10 @@ def _compute_local_treatment_effects_core(
185185

186186
# Compute outcome indicators (different for LDTE vs LPTE)
187187
if use_intervals:
188-
bi = (Y[i] < locations) * 1
188+
bi = (Y[i] <= locations) * 1
189189
bi = bi[1:] - bi[:-1] # Convert to interval probabilities
190190
else:
191-
bi = Y[i] <= locations
191+
bi = (Y[i] <= locations) * 1
192192

193193
xi_t[i] = ((1 - 1 / w_s) * y_t_mu[i] - y_c_mu[i] + bi / w_s) - beta * (
194194
(1 - 1 / w_s) * d_t_eta[i] - d_c_eta[i] + D[i] / w_s

0 commit comments

Comments
 (0)