fix: imports

mlr-org · May 1, 2024 · 3bd95b0 · 3bd95b0
1 parent 170d650
commit 3bd95b0
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -53,7 +53,6 @@ import(mlr3)
 import(mlr3misc)
 import(mlr3tuning)
 import(paradox)
-import(rush)
 import(spacefillr)
 importFrom(R6,R6Class)
 importFrom(stats,dnorm)

diff --git a/R/zzz.R b/R/zzz.R
@@ -9,7 +9,6 @@
 #' @import lgr
 #' @import mlr3
 #' @import mlr3tuning
-#' @import rush
 #' @importFrom stats setNames runif dnorm pnorm quantile rexp
 #' @useDynLib mlr3mbo c_sms_indicator c_eps_indicator
 "_PACKAGE"

diff --git a/tests/testthat/test_OptimizerADBO.R b/tests/testthat/test_OptimizerADBO.R
@@ -3,7 +3,7 @@ test_that("adbo optimizer works", {
   skip_if_not_installed("rush")
   flush_redis()
 
-  rush_plan(n_workers = 2)
+  rush::rush_plan(n_workers = 2)
   instance = oi_async(
     objective = OBJ_2D,
     search_space = PS_2D,

diff --git a/tests/testthat/test_TunerADBO.R b/tests/testthat/test_TunerADBO.R
@@ -33,7 +33,7 @@ test_that("adbo works with transformation functions", {
     minsplit  = to_tune(2, 128, logscale = TRUE),
     cp        = to_tune(1e-04, 1e-1, logscale = TRUE))
 
-  rush_plan(n_workers = 2)
+  rush::rush_plan(n_workers = 2)
   instance = ti_async(
     task = tsk("pima"),
     learner = learner,
@@ -60,7 +60,7 @@ test_that("search works with dependencies", {
     cp        = to_tune(1e-04, 1e-1),
     keep_model = to_tune())
 
-  rush_plan(n_workers = 2)
+  rush::rush_plan(n_workers = 2)
   instance = ti_async(
     task = tsk("pima"),
     learner = learner,
@@ -92,7 +92,7 @@ test_that("adbo works with branching", {
     "branch.selection" = to_tune(c("rpart", "debug"))
   )
 
-  rush_plan(n_workers = 2)
+  rush::rush_plan(n_workers = 2)
   instance = ti_async(
     task = tsk("pima"),
     learner = graph_learner,