From 1318d54b310e649a5022949b77885041a9e99860 Mon Sep 17 00:00:00 2001 From: Daniel Date: Sun, 1 Sep 2024 18:17:35 +0200 Subject: [PATCH] fix --- R/find_parameters.R | 2 +- tests/testthat/test-weightit.R | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/R/find_parameters.R b/R/find_parameters.R index cc9f0e4b8..7b6e78833 100644 --- a/R/find_parameters.R +++ b/R/find_parameters.R @@ -177,7 +177,7 @@ find_parameters.brmultinom <- find_parameters.multinom find_parameters.multinom_weightit <- function(x, flatten = FALSE, ...) { params <- stats::coef(x) resp <- gsub("(.*)~(.*)", "\\1", names(params)) - pars <- gsub("(.*)~(.*)", "\\2", names(params))[resp == resp[1]] + pars <- list(conditional = gsub("(.*)~(.*)", "\\2", names(params))[resp == resp[1]]) if (flatten) { unique(unlist(pars, use.names = FALSE)) diff --git a/tests/testthat/test-weightit.R b/tests/testthat/test-weightit.R index 9cc62b97f..b3cfcfc67 100644 --- a/tests/testthat/test-weightit.R +++ b/tests/testthat/test-weightit.R @@ -14,10 +14,11 @@ w.out <- WeightIt::weightit( ) set.seed(123) fit3 <- WeightIt::lm_weightit( - re78 ~ treat + age + educ, data = lalonde, + re78 ~ treat + age + educ, + data = lalonde, weightit = w.out, vcov = "FWB", - R = 50, #should use way more + R = 50, # should use way more fwb.args = list(wtype = "mammen") )