closes #197

DeclareDesign · Oct 21, 2018 · 806cc11 · 806cc11
1 parent 00be7c0
commit 806cc11
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/R/pretest_posttest_designer.R b/R/pretest_posttest_designer.R
@@ -45,16 +45,16 @@ pretest_posttest_designer <- function(N = 100,
       u_t2 = rnorm(N, rho * u_t1, sqrt(1 - rho^2))*sd_2,
       Y_t1 = u_t1
     )
-
+    
     potential_outcomes <- declare_potential_outcomes(Y_t2 ~ u_t2 + ate * Z)
 
     # I: Inquiry
     estimand <- declare_estimand(ATE = mean(Y_t2_Z_1 - Y_t2_Z_0))
 
     # D: Data Strategy
     assignment <- declare_assignment()
-    report     <- declare_assignment(m = round(N * (1 - attrition_rate)),
-                                    assignment_variable = R)
+    report     <- declare_assignment(prob = 1 - attrition_rate,
+                                     assignment_variable = R)
     reveal_t2 <- declare_reveal(Y_t2) 
     manipulation <- declare_step(difference = (Y_t2 - Y_t1), handler = fabricate)