include pd values in output, much easier to understand

ropensci · Jan 18, 2024 · 919f655 · 919f655
1 parent b3d2e29
commit 919f655
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/R/orsf_vint.R b/R/orsf_vint.R
@@ -113,19 +113,15 @@ orsf_vint <- function(object,
  pd_split <- split(pd, by = split_vars)
 
  # for cran
- . <- score <- var_1_value <- var_2_value <- NULL
+ . <- score <- var_1_value <- var_2_value <- pd_values <- NULL
 
  pd_scores <- vapply(
   pd_split,
   function(dt){
-   mean(
+   collapse::fmean(
     c(
-     collapse::fsd(
-      dt[, .(vi = collapse::fsd(mean)), by = var_1_value][["vi"]]
-     ),
-     collapse::fsd(
-      dt[, .(vi = collapse::fsd(mean)), by = var_2_value][["vi"]]
-     )
+     collapse::fsd(dt[, .(vi = collapse::fsd(mean)), by = var_1_value][["vi"]]),
+     collapse::fsd(dt[, .(vi = collapse::fsd(mean)), by = var_2_value][["vi"]])
     )
    )
   },