Relax lm_forest invariance (#1458)

grf-labs · Oct 3, 2024 · 8df5a52 · 8df5a52
1 parent 0823d49
commit 8df5a52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/r-package/grf/tests/testthat/test_lm_forest.R b/r-package/grf/tests/testthat/test_lm_forest.R
@@ -51,7 +51,7 @@ test_that("lm_forest with dummy W = multi arm causal forest", {
 
 test_that("lm_forest gradient.weights option works as expected", {
   n <- 250
-  p <- 5
+  p <- 1
   K <- 2
   X <- matrix(rnorm(n * p), n, p)
   W <- matrix(runif(n * K), n, K)