refactor: add key to result

mlr-org · Jul 19, 2024 · b9521e9 · b9521e9
1 parent fbcfc6d
commit b9521e9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/Optimizer.R b/R/Optimizer.R
@@ -166,7 +166,9 @@ Optimizer = R6Class("Optimizer",
 assign_result_default = function(inst) {
   assert_r6(inst, "OptimInstance")
   res = inst$archive$best()
-  xdt = res[, inst$search_space$ids(), with = FALSE]
+  ids = inst$search_space$ids()
+  if ("keys" %in% names(res)) ids = c(ids, "keys")
+  xdt = res[, ids, with = FALSE]
 
   if (inherits(inst, "OptimInstanceBatchMultiCrit") || inherits(inst, "OptimInstanceAsyncMultiCrit")) {
     ydt = res[, inst$archive$cols_y, with = FALSE]