pass through the trimming threshold to the functional implementation

Author: MalteKurz

tests/testthat/test-double_ml_iivm.R

@@ -10,14 +10,14 @@ if (on_cran) {
     learner = "rpart",
     dml_procedure = "dml2",
     score = "LATE",
-    trimming_threshold = c(0.0001),
+    trimming_threshold = c(1e-5),
     stringsAsFactors = FALSE)
 } else {
   test_cases = expand.grid(
     learner = "cv_glmnet",
     dml_procedure = c("dml1", "dml2"),
     score = "LATE",
-    trimming_threshold = c(0.0001),
+    trimming_threshold = c(1e-5),
     stringsAsFactors = FALSE)
 }
 
@@ -35,7 +35,8 @@ patrick::with_parameters_test_that("Unit tests for IIVM:",
       ml_g = learner_pars$ml_g$clone(),
       ml_m = learner_pars$ml_m$clone(),
       ml_r = learner_pars$ml_r$clone(),
-      dml_procedure = dml_procedure, score = score)
+      dml_procedure = dml_procedure, score = score,
+      trimming_threshold = trimming_threshold)
     theta = iivm_hat$coef
     se = iivm_hat$se
 
@@ -45,7 +46,8 @@ patrick::with_parameters_test_that("Unit tests for IIVM:",
       n_folds = 5, smpls = iivm_hat$smpls,
       all_preds = iivm_hat$all_preds,
       score = score,
-      bootstrap = "normal", n_rep_boot = n_rep_boot)$boot_coef
+      bootstrap = "normal", n_rep_boot = n_rep_boot,
+      trimming_threshold = trimming_threshold)$boot_coef
 
     set.seed(3141)
     double_mliivm_obj = DoubleMLIIVM$new(

tests/testthat/test-double_ml_irm_user_score.R

@@ -25,15 +25,15 @@ if (on_cran) {
     learner = "regr.rpart",
     learner_m = "classif.rpart",
     dml_procedure = "dml2",
-    trimming_threshold = 0.0001,
+    trimming_threshold = 1e-5,
     stringsAsFactors = FALSE)
   test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
 } else {
   test_cases = expand.grid(
     learner = "regr.glmnet",
     learner_m = "classif.glmnet",
     dml_procedure = c("dml1", "dml2"),
-    trimming_threshold = c(0.0001, 0.01),
+    trimming_threshold = c(1e-5, 0.01),
     stringsAsFactors = FALSE)
   test_cases[".test_name"] = apply(test_cases, 1, paste, collapse = "_")
 }