fix: no visible binding

mlr-org · Jun 29, 2024 · 6b97676 · 6b97676
1 parent 37baad3
commit 6b97676
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/NAMESPACE b/NAMESPACE
@@ -16,3 +16,4 @@ import(mlr3tuning)
 import(paradox)
 importFrom(R6,R6Class)
 importFrom(utils,bibentry)
+importFrom(utils,head)
diff --git a/R/OptimizerBatchHyperband.R b/R/OptimizerBatchHyperband.R
@@ -159,7 +159,7 @@ OptimizerBatchHyperband = R6Class("OptimizerBatchHyperband",
           # promote configurations of previous batch
           if (s != s_max) {
             archive = inst$archive
-            data = archive$data[batch_nr == archive$n_batch, ]
+            data = archive$data[archive$n_batch, , on = "batch_nr"]
             minimize = ifelse(archive$codomain$maximization_to_minimization == -1, TRUE, FALSE)
 
             # for each bracket, promote configurations of previous stage

diff --git a/R/zzz.R b/R/zzz.R
@@ -7,6 +7,7 @@
 #' @import mlr3tuning
 #' @import bbotk
 #' @importFrom R6 R6Class
+#' @importFrom utils head
 "_PACKAGE"
 
 register_bbotk = function() {