fix new_target condition

mlr-org · Feb 23, 2025 · fe42407 · fe42407
1 parent 083def4
commit fe42407
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/PipeOpTrafo.R b/R/PipeOpTrafo.R
@@ -371,7 +371,7 @@ PipeOpTargetMutate = R6Class("PipeOpTargetMutate",
 
     .transform = function(task, phase) {
       new_target = self$param_set$values$trafo(task$data(cols = task$target_names))
-      if (!is.data.frame(new_target) || !is.matrix(new_target)) {
+      if (!is.data.frame(new_target) && !is.matrix(new_target)) {
         stopf("Hyperparameter 'trafo' must be a function returning a 'data.frame', 'data.table', or 'matrix', not '%s'.", class(new_target))
       }
       task$cbind(new_target)