From 3bd95b0d7218c1b8d39ad7da33d4f817b3190655 Mon Sep 17 00:00:00 2001
From: be-marc <marcbecker@posteo.de>
Date: Wed, 1 May 2024 08:27:28 +0200
Subject: [PATCH] fix: imports

---
 NAMESPACE                           | 1 -
 R/zzz.R                             | 1 -
 tests/testthat/test_OptimizerADBO.R | 2 +-
 tests/testthat/test_TunerADBO.R     | 6 +++---
 4 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/NAMESPACE b/NAMESPACE
index abd5e315..4ab0552c 100644
--- 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
index 6ed1197e..a0e1e22d 100644
--- 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
index 8aecb5ea..6e8a73a1 100644
--- 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
index 40e14c70..2bc24ba2 100644
--- 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,