split data.frame without forcing it into data.table

bips-hb · Feb 21, 2025 · 9b1e261 · 9b1e261
1 parent ad6fd47
commit 9b1e261
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/R/impute.R b/R/impute.R
@@ -114,10 +114,7 @@ impute <- function(x,
                                 evidence = list(x), 
                                 round = list(round),
                                 forge_args))
-    x_synth <- as.data.table(x_synth)
-    x_synth[, idx := rep(1:m, nrow(x))]
-    x_imputed <- split(x_synth, by = "idx")
-    x_imputed <- lapply(x_imputed, function(x) x[, idx := NULL])
+    x_imputed <- split(x_synth, rep(1:m, nrow(x)))
   }
   x_imputed
 }