Skip to content

Commit 0edfc12

Browse files
committed
fix another post-merge issue
1 parent 64584b5 commit 0edfc12

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

R/double_ml_plr.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -524,8 +524,8 @@ DoubleMLPLR = R6Class("DoubleMLPLR",
524524
d = self$data$data_model[[self$data$treat_col]]
525525
y = self$data$data_model[[self$data$y_col]]
526526

527-
psi_a = -(d - m_hat) * (d - m_hat)
528-
psi_b = (d - m_hat) * (y - l_hat)
527+
psi_a = -(d - m_hat$preds) * (d - m_hat$preds)
528+
psi_b = (d - m_hat$preds) * (y - l_hat$preds)
529529
theta_initial = -mean(psi_b, na.rm = TRUE) / mean(psi_a, na.rm = TRUE)
530530

531531
data_aux = data.table(self$data$data_model,

0 commit comments

Comments
 (0)