chore: export with_rng_state

NAMESPACE

@@ -10,6 +10,7 @@ export(rush_available)
 export(rush_config)
 export(rush_plan)
 export(start_worker)
+export(with_rng_state)
 export(worker_loop_callr)
 export(worker_loop_default)
 import(checkmate)

R/Rush.R

@@ -1338,11 +1338,12 @@ Rush = R6::R6Class("Rush",
 
       # find globals
       if (!is.null(globals)) {
+        global_names = if (!is.null(names(globals))) names(globals) else globals
         globals = set_names(map(globals, function(global) {
           value = get(global, envir = parent.frame(), inherits = TRUE)
           if (is.null(value)) stopf("Global `%s` not found", global)
           value
-        }), globals)
+        }), global_names)
       }
 
       # arguments needed for initializing RushWorker

R/helper.R

@@ -23,6 +23,25 @@ get_hostname = function() {
   host[1]
 }
 
+
+#' @title Set RNG Sate before Running a Function
+#'
+#' @description
+#' This function sets the rng state before running a function.
+#'
+#' @param fun (`function`)\cr
+#' Function to run.
+#' @param args (`list`)\cr
+#' Arguments to pass to `fun`.
+#' @param seed (`integer`)\cr
+#' RNG state to set before running `fun`.
+#'
+#' @export
+with_rng_state = function(fun, args, seed) {
+  if (!is.null(seed)) assign(".Random.seed", seed, envir = globalenv())
+  mlr3misc::invoke(fun, .args = args)
+}
+
 is_lecyer_cmrg_seed = function(seed) {
   is.numeric(seed) && length(seed) == 7L && all(is.finite(seed)) && (seed[1] %% 10000L == 407L)
 }
@@ -59,13 +78,6 @@ safe_bin_to_object = function(bin) {
   redux::bin_to_object(bin)
 }
 
-# runs code with a specific rng state
-with_rng_state = function(fun, args, seed) {
-  if (!is.null(seed)) assign(".Random.seed", seed, envir = globalenv())
-  mlr3misc::invoke(fun, .args = args)
-}
-
-
 is_retriable = function(task) {
   if (is.null(task$max_retries)) return(FALSE)
   assert_int(task$max_retries)