Update progress bar to be current and run in caller env

openjusticeok · Apr 15, 2024 · 2bf2f44 · 2bf2f44
1 parent b7fbfa6
commit 2bf2f44
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/R/ojo_collect.R b/R/ojo_collect.R
@@ -110,13 +110,14 @@ collect_data <- function(.con, .data, n_results, .silent) {
     }
   }
 
-  if (!.silent) {
-    cli::cli_progress_done()
-  }
-
   return(res)
 }
 
 init_progress_bar <- function(n_results) {
-  cli::cli_progress_bar(name = "dl_pb", total = n_results, clear = FALSE)
+  cli::cli_progress_bar(
+    type = "iterator",
+    total = n_results,
+    clear = FALSE,
+    .envir = rlang::caller_env()
+  )
 }