fix: add rush parameter to auto tuner

R/AutoTuner.R

@@ -159,6 +159,7 @@ AutoTuner = R6Class("AutoTuner",
 
       ia$check_values = assert_flag(check_values)
       ia$callbacks = assert_callbacks(as_callbacks(callbacks))
+      if (!is.null(rush)) ia$rush = assert_class(rush, "Rush")
       self$instance_args = ia
 
       super$initialize(

R/ObjectiveTuning.R

@@ -60,8 +60,8 @@ ObjectiveTuning = R6Class("ObjectiveTuning",
       self$measures = assert_measures(as_measures(measures), task = self$task, learner = self$learner)
       self$store_models = assert_flag(store_models)
       self$store_benchmark_result = assert_flag(store_benchmark_result) || self$store_models
-
       self$callbacks = assert_callbacks(as_callbacks(callbacks))
+
       self$default_values = self$learner$param_set$values
 
       super$initialize(

R/auto_tuner.R

@@ -21,6 +21,7 @@
 #' @template param_store_models
 #' @template param_check_values
 #' @template param_callbacks
+#' @template param_rush
 #'
 #' @export
 #' @examples
@@ -45,7 +46,8 @@ auto_tuner = function(
   store_benchmark_result = TRUE,
   store_models = FALSE,
   check_values = FALSE,
-  callbacks = NULL
+  callbacks = NULL,
+  rush = NULL
   ) {
   terminator = terminator %??% terminator_selection(term_evals, term_time)
 
@@ -60,5 +62,6 @@ auto_tuner = function(
     store_benchmark_result = store_benchmark_result,
     store_models = store_models,
     check_values = check_values,
-    callbacks = callbacks)
+    callbacks = callbacks,
+    rush = rush)
 }

tests/testthat/test_auto_tuner.R

@@ -20,3 +20,46 @@ test_that("auto_tuner function works", {
   expect_class(at, "AutoTuner")
   expect_class(at$instance_args$terminator, "TerminatorCombo")
 })
+
+test_that("async auto tuner works", {
+  skip_if_not_installed("rush")
+  flush_redis()
+
+  rush_plan(n_workers = 2)
+
+  at = auto_tuner(
+    tuner = tnr("async_random_search"),
+    learner = lrn("classif.rpart", cp = to_tune(0.01, 0.1)),
+    resampling = rsmp("cv", folds = 3),
+    measure = msr("classif.ce"),
+    terminator = trm("evals", n_evals = 3)
+  )
+
+  expect_class(at, "AutoTuner")
+  at$train(tsk("pima"))
+
+  expect_class(at$tuning_instance, "TuningInstanceAsyncSingleCrit")
+})
+
+test_that("async auto tuner works with rush controller", {
+  skip_if_not_installed("rush")
+  flush_redis()
+
+  rush_plan(n_workers = 2)
+  rush = rsh(network_id = "tuning_network")
+
+  at = auto_tuner(
+    tuner = tnr("async_random_search"),
+    learner = lrn("classif.rpart", cp = to_tune(0.01, 0.1)),
+    resampling = rsmp("cv", folds = 3),
+    measure = msr("classif.ce"),
+    terminator = trm("evals", n_evals = 3),
+    rush = rush
+  )
+
+  expect_class(at, "AutoTuner")
+  expect_class(at$instance_args$rush, "Rush")
+  at$train(tsk("pima"))
+
+  expect_class(at$tuning_instance, "TuningInstanceAsyncSingleCrit")
+})