Merge pull request #34 from Merck/wangben718-patch-1

Update collect.R
Merck · Jul 30, 2024 · c3ead88 · c3ead88
2 parents c47d53e + b4a9413
commit c3ead88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/R/collect.R b/R/collect.R
@@ -178,7 +178,9 @@ collect_baseline <- function(
   if (!is.null(par_var_group)) {
     vargroup <- pop[[par_var_group]]
     stopifnot(inherits(vargroup, c("factor", "character")))
-    vargroup <- factor(vargroup, exclude = NULL)
+    if (any(c("character") %in% class_var)) {
+      vargroup <- factor(vargroup, exclude = NULL)
+    }
 
     if (all(is.na(vargroup))) {
       levels(vargroup) <- c(levels(vargroup), title["missing"])
@@ -211,7 +213,9 @@ collect_baseline <- function(
       group_lower <- pop_lower[[pop_group]]
       varlower <- pop_lower[[par_var_lower]]
       stopifnot(inherits(varlower, c("factor", "character")))
+      if (any(c("character") %in% class_var)) {
       varlower <- factor(varlower, exclude = NULL)
+      }
 
       if (length(varlower) > 0) {
         lower <- metalite::n_subject(id_lower, group_lower, par = varlower)