Skip to content

Commit 48dc1f8

Browse files
committed
remove unemployment task from test
1 parent 8f5651e commit 48dc1f8

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/testthat/test_single_step.R

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,18 @@
11
test_that("single-step surv", {
2-
task = mlr_tasks$get("unemployment")
2+
task = mlr_tasks$get("gbcs")
33
learner = mlr_learners$get("surv.kaplan")
44

5-
train_task = task$clone()$filter(1:3000)
5+
train_task = task$clone()$filter(1:600)
66
learner$train(train_task)
7-
newdata = remove_named(task$clone()$filter(3001:3343)$data(), task$target_names)
7+
newdata = remove_named(task$clone()$filter(601:686)$data(), task$target_names)
88
p = learner$predict_newdata(task = train_task, newdata = newdata)
99

1010
p = as.data.table(p)
11-
expect_data_table(p, nrow = 343L)
11+
expect_data_table(p, nrow = 86L)
1212
expect_true(allMissing(p$time))
1313
expect_true(allMissing(p$status))
1414
})
1515

16-
1716
test_that("single-step dens", {
1817
task = tsk("faithful")
1918
learner = lrn("dens.kde")

0 commit comments

Comments
 (0)