Fixes

ngreifer · Aug 20, 2024 · 6dd448c · 6dd448c
1 parent 6cdb3c8
commit 6dd448c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/R/weightit2cbps.R b/R/weightit2cbps.R
@@ -140,7 +140,7 @@
 #'                 method = "cbps"))
 #' summary(W3)
 #' cobalt::bal.tab(W3)
-#'
+#' \donttest{
 #' #Longitudinal treatments
 #' data("msmdata")
 #' (W4 <- weightitMSM(list(A_1 ~ X1_0 + X2_0,
@@ -150,6 +150,7 @@
 #'                    method = "cbps"))
 #' summary(W4)
 #' cobalt::bal.tab(W4)
+#' }
 NULL
 
 weightit2cbps <- function(covs, treat, s.weights, estimand, focal, subset,

diff --git a/R/weightit2gbm.R b/R/weightit2gbm.R
@@ -242,7 +242,7 @@ weightit2gbm <- function(covs, treat, s.weights, estimand, focal, subset,
     chk::chk_string(criterion)
   }
 
-  available.criteria <- cobalt::available.stats(treat.type)
+  available.criteria <- cobalt::available.stats(switch(treat.type, "multi-category" = "multi", treat.type))
 
   if (startsWith(criterion, "es.")) {
     subbed.crit <- sub("es.", "smd.", criterion, fixed = TRUE)

diff --git a/man/method_cbps.Rd b/man/method_cbps.Rd