replace vroom for write and replace vroom in DESCRIPTION

Author: rok-cesnovar

DESCRIPTION

@@ -29,7 +29,7 @@ Imports:
     posterior (>= 0.1.0),
     processx,
     R6 (>= 2.4.0),
-    vroom
+    data.table
 Suggests: 
     bayesplot,
     knitr,

R/data.R

@@ -162,19 +162,22 @@ process_fitted_params <- function(fitted_params) {
         paths <- file.path(tempdir(), paths)
         chain <- 1
         for (path in paths) {
-          chain_draws <- posterior::as_draws_df(posterior::subset_draws(draws, chain = chain))
-          colnames(chain_draws) <- unrepair_variable_names(variables)
+          chain_draws <- posterior::subset_draws(draws, chain = chain)
+          unname(chain_draws)         
           write(
             paste0("# num_samples = ", iterations),
+            file = path
+          )
+          write(
+            paste0(unrepair_variable_names(variables), collapse = ","),
             file = path,
-            append = FALSE
+            append = TRUE
           )
-          vroom::vroom_write(
+          utils::write.table(
             chain_draws,
-            delim = ",",
-            path = path,
-            col_names = TRUE,
-            progress = FALSE,
+            file = path,
+            sep = ",",
+            col.names = FALSE,
             append = TRUE
           )
           chain <- chain + 1

R/read_csv.R

@@ -121,8 +121,6 @@ read_cmdstan_csv <- function(files,
   col_types <- NULL
   col_select <- NULL
   not_matching <- c()
-  vroom_warnings <- 0
-
   for (output_file in files) {
     if (is.null(metadata)) {
       metadata <- read_csv_metadata(output_file)

tests/testthat/test-data.R

@@ -81,14 +81,9 @@ test_that("process_fitted_params() works if output_files in fit do not exist", {
   chain <- 1
   for(file in new_files) {
     suppressWarnings(
-      tmp_file_gq <- vroom::vroom(
-        file,
-        comment = "#",
-        delim = ',',
-        trim_ws = TRUE,
-        altrep = FALSE,
-        progress = FALSE,
-        skip = 1)
+      tmp_file_gq <- data.table::fread(
+        cmd = paste0("grep -v '^#' ", file)
+      )
     )
     tmp_file_gq <- posterior::as_draws_array(tmp_file_gq)
     expect_equal(