Merge branch 'main' of https://github.com/mlr-org/mlr3proba

Author: bblodfon

R/LearnerSurvKaplan.R

@@ -49,6 +49,10 @@ LearnerSurvKaplan = R6Class("LearnerSurvKaplan",
     #'
     #' @return `character(0)`.
     selected_features = function() {
+      if (is.null(self$model)) {
+        stopf("No model stored")
+      }
+
       character()
     }
   ),

tests/testthat/helper.R

@@ -50,3 +50,4 @@ gen_cif = function(n = 20, n_events = 2, n_times = 20) {
 
   cif_list
 }
+

tests/testthat/test_mlr_learners_surv_kaplan.R

@@ -14,3 +14,12 @@ test_that("single prediction", {
   learner$train(task)
   expect_prediction_surv(learner$predict(task, task$row_ids[1]))
 })
+
+test_that("importance/selected", {
+  learner = lrn("surv.kaplan")
+  expect_error(learner$importance(), "No model stored")
+  expect_error(learner$selected_features(), "No model stored")
+  learner$train(tsk("rats"))
+  expect_character(learner$selected_features(), len = 0)
+  expect_named(learner$importance())
+})