fixed the data.table indexing passed to impute

tlverse · Oct 4, 2023 · b75abeb · b75abeb
1 parent 7433178
commit b75abeb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/process_data.R b/R/process_data.R
@@ -134,7 +134,8 @@ process_data <- function(data, nodes, column_names, flag = TRUE,
       nodes$covariates <- c(nodes$covariates, missing_indicator_vars)
     }
     # impute covariates
-    imputed <- impute(data[, missing_covar_cols, with = FALSE])
+    data_missing_covars <- data[, missing_covar_cols, with = FALSE]
+    imputed <- impute(data.frame(data_missing_covars))
 
     # update data
     set(data, , missing_covar_cols, imputed)