@@ -35,7 +35,9 @@ patrick::with_parameters_test_that("Unit tests for IIVM:",
3535 ml_g = learner_pars $ ml_g $ clone(),
3636 ml_m = learner_pars $ ml_m $ clone(),
3737 ml_r = learner_pars $ ml_r $ clone(),
38- dml_procedure = dml_procedure , score = score )
38+ dml_procedure = dml_procedure ,
39+ trimming_threshold = trimming_threshold ,
40+ score = score )
3941 theta = iivm_hat $ coef
4042 se = iivm_hat $ se
4143
@@ -66,8 +68,8 @@ patrick::with_parameters_test_that("Unit tests for IIVM:",
6668 boot_theta_obj = double_mliivm_obj $ boot_coef
6769
6870 # at the moment the object result comes without a name
69- expect_equal(theta , theta_obj , tolerance = 1e-4 )
70- expect_equal(se , se_obj , tolerance = 1e-4 )
71- expect_equal(as.vector(boot_theta ), as.vector(boot_theta_obj ), tolerance = 1e-4 )
71+ expect_equal(theta , theta_obj , tolerance = 1e-8 )
72+ expect_equal(se , se_obj , tolerance = 1e-8 )
73+ expect_equal(as.vector(boot_theta ), as.vector(boot_theta_obj ), tolerance = 1e-8 )
7274 }
7375)
0 commit comments