add mirai in tests

Author: bblodfon

R/LearnerSurvKaplan.R

@@ -19,22 +19,50 @@ LearnerSurvKaplan = R6Class("LearnerSurvKaplan",
         id = "surv.kaplan",
         predict_types = c("crank", "distr"),
         feature_types = c("logical", "integer", "numeric", "character", "factor", "ordered"),
-        properties = "missings",
+        properties = c("missings", "importance", "selected_features"),
         packages = c("survival", "distr6"),
         label = "Kaplan-Meier Estimator",
         man = "mlr3proba::mlr_learners_surv.kaplan"
       )
+    },
+
+    #' @description
+    #' All features have a score of `0` for this learner.
+    #' #' This method exists solely for compatibility with the `mlr3` ecosystem,
+    #' as this learner is used as a fallback for other survival learners that
+    #' require an `importance()` method.
+    #'
+    #' @return Named `numeric()`.
+    importance = function() {
+      if (is.null(self$model)) {
+        stopf("No model stored")
+      }
+
+      fn = self$model$features
+      named_vector(fn, 0)
+    },
+
+    #' @description
+    #' Selected features are always the empty set for this learner.
+    #' This method is implemented only for compatibility with the `mlr3` API,
+    #' as this learner does not perform feature selection.
+    #'
+    #' @return `character(0)`.
+    selected_features = function() {
+      character()
     }
   ),
 
   private = list(
     .train = function(task) {
-      invoke(survival::survfit, formula = task$formula(1), data = task$data())
+      list(model = invoke(survival::survfit, formula = task$formula(1),
+                          data = task$data(cols = task$target_names)),
+           features = task$feature_names) # keep for importance
     },
 
     .predict = function(task) {
-      times = self$model$time
-      surv = matrix(rep(self$model$surv, task$nrow), ncol = length(times),
+      times = self$model$model$time
+      surv = matrix(rep(self$model$model$surv, task$nrow), ncol = length(times),
                     nrow = task$nrow, byrow = TRUE)
 
       .surv_return(times = times, surv = surv)

man/mlr_learners_surv.kaplan.Rd

@@ -1,3 +1,6 @@
 lg = lgr::get_logger("mlr3")
 old_threshold = lg$threshold
 lg$set_threshold("warn")
+
+mirai::daemons(0, .compute = "mlr3_encapsulation")
+mirai::daemons(1, .compute = "mlr3_encapsulation")

tests/testthat/setup.R

@@ -1 +1,3 @@
 lg$set_threshold(old_threshold)
+
+mirai::daemons(0, .compute = "mlr3_encapsulation")