We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1208350 commit 2fc68ccCopy full SHA for 2fc68cc
1 file changed
dte_adj/util.py
@@ -185,10 +185,10 @@ def _compute_local_treatment_effects_core(
185
186
# Compute outcome indicators (different for LDTE vs LPTE)
187
if use_intervals:
188
- bi = (Y[i] < locations) * 1
+ bi = (Y[i] <= locations) * 1
189
bi = bi[1:] - bi[:-1] # Convert to interval probabilities
190
else:
191
- bi = Y[i] <= locations
192
193
xi_t[i] = ((1 - 1 / w_s) * y_t_mu[i] - y_c_mu[i] + bi / w_s) - beta * (
194
(1 - 1 / w_s) * d_t_eta[i] - d_c_eta[i] + D[i] / w_s
0 commit comments