minor speed improvements

vincentarelbundock · Sep 17, 2023 · de58753 · de58753
1 parent d74d94e
commit de58753
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/R/comparisons.R b/R/comparisons.R
@@ -471,7 +471,8 @@ comparisons <- function(model,
         se <- do.call("get_se_delta", args)
         J <- attr(se, "jacobian")
         attr(se, "jacobian") <- NULL
-        mfx$std.error <- as.numeric(se)
+        data.table::setDT(mfx)
+        mfx[, "std.error" := drop(se)]
         draws <- NULL
 
     # no standard error

diff --git a/R/get_modeldata.R b/R/get_modeldata.R
@@ -70,7 +70,7 @@ get_modeldata <- function(model, additional_variables = FALSE, modeldata = NULL,
         out <- evalup(attr(model, "call")$data)
     }
 
-    out <- as.data.frame(out)
+    data.table::setDF(out)
     out <- set_variable_class(out, model = model)
     return(out)
 }

diff --git a/R/marginal_means.R b/R/marginal_means.R
@@ -348,10 +348,10 @@ marginal_means <- function(model,
         modeldata = modeldata)
     args <- c(args, list(...))
     args[["equivalence"]] <- NULL
-    mm <- do.call(get_marginalmeans, args)
+    out <- do.call(get_marginalmeans, args)
 
     # we want consistent output, regardless of whether `data.table` is installed/used or not
-    out <- as.data.frame(mm)
+    data.table::setDF(out)
 
     # standard errors via delta method
     if (!vcov_false) {