upkeep with mlr3

Author: sebffischer

R/GraphLearner.R

@@ -272,6 +272,13 @@ GraphLearner = R6Class("GraphLearner", inherit = Learner,
         if (!some_pipeops_validate) {
           lg$warn("GraphLearner '%s' specifies a validation set, but none of its Learners use it.", self$id)
         }
+      } else {
+        # otherwise the pipeops will preprocess this unnecessarily
+        if (!is.null(task$internal_valid_task)) {
+          prev_itv = task$internal_valid_task
+          on.exit({task$internal_valid_task = prev_itv}, add = TRUE)
+          task$internal_valid_task = NULL
+        }
       }
 
       on.exit({self$graph$state = NULL})

R/pipeline_branch.R

@@ -91,10 +91,9 @@ pipeline_branch = function(graphs, prefix_branchops = "", prefix_paths = FALSE)
     pmap(list(
       src_id = branch_id, dst_id = gin$op.id,
       src_channel = branch_chan, dst_channel = gin$channel.name),
-      graph$add_edge)
+    graph$add_edge)
   })
   graph
 }
 
 mlr_graphs$add("branch", pipeline_branch)
-