...

mlr-org · Dec 10, 2024 · 50a26df · 50a26df
1 parent 15f9e76
commit 50a26df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/testthat/test_CallbackEvaluation.R b/tests/testthat/test_CallbackEvaluation.R
@@ -91,14 +91,15 @@ test_that("writing to learner$state works", {
   walk(rr$learners, function(learner) {
     expect_equal(learner$state$test, 1)
   })
+  expect_null(rr$data_extra)
 
   # benchmark result
   design = benchmark_grid(task, learner, resampling)
   bmr = benchmark(design, callbacks = callback)
-
   walk(bmr$score()$learner, function(learner) {
     expect_equal(learner$state$test, 1)
   })
+  expect_null(bmr$data_extra)
 })
 
 test_that("writing to data_extra works", {