...

mlr-org · Nov 4, 2024 · e93a9f2 · e93a9f2
1 parent a9ce0f0
commit e93a9f2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/testthat/test_TaskRegr.R b/tests/testthat/test_TaskRegr.R
@@ -1,9 +1,9 @@
-test_that("Basic ops on BostonHousing task", {
+test_that("Basic ops on Ames Housing task", {
   task = tsk("ames_housing")
   expect_task(task)
   expect_task_supervised(task)
   expect_task_regr(task)
-  expect_equal(task$target_names, "cmedv")
+  expect_equal(task$target_names, "Sale_Price")
 
   f = task$formula()
   expect_class(f, "formula")

diff --git a/tests/testthat/test_predict.R b/tests/testthat/test_predict.R
@@ -51,7 +51,7 @@ test_that("missing predictions are handled gracefully / regr", {
 
 test_that("predict_newdata with weights (#519)", {
   task = tsk("ames_housing")
-  task$set_col_roles("Bsmt_Half_Bath", "weight")
+  task$set_col_roles("Bedroom_AbvGr", "weight")
   learner = lrn("regr.featureless")
   learner$train(task)
   expect_prediction(learner$predict(task))