fix tests

mlr-org · Feb 4, 2025 · fc90fcc · fc90fcc
1 parent df25eab
commit fc90fcc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/worker.R b/R/worker.R
@@ -22,6 +22,8 @@ learner_train = function(learner, task, train_row_ids = NULL, test_row_ids = NUL
       model = marshal_model(model, inplace = TRUE)
     }
 
+    learner$state$model = model
+
     # Extract internal valid scores and tuned values if applicable.
     internal_valid_scores = if (!is.null(get0("validate", learner)) &&
       exists(".extract_internal_valid_scores", get_private(learner))) {