fix typo (#491)

grf-labs · Aug 21, 2019 · 9b7cff9 · 9b7cff9
1 parent 3f75a67
commit 9b7cff9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/r-package/grf/R/average_treatment_effect.R b/r-package/grf/R/average_treatment_effect.R
@@ -95,7 +95,7 @@ average_treatment_effect <- function(forest,
   clusters <- if (cluster.se) {
     forest$clusters
   } else {
-    1:length(forest$Y)
+    1:length(forest$Y.orig)
   }
   observation.weight <- observation_weights(forest)