Update R/PipeOpTaskPreprocTorch.R

mlr-org · Jan 23, 2024 · fbe093f · fbe093f
1 parent 9dc507c
commit fbe093f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/R/PipeOpTaskPreprocTorch.R b/R/PipeOpTaskPreprocTorch.R
@@ -399,7 +399,11 @@ create_ps = function(fn) {
   args = list()
 
   args = set_names(map(param_names, function(pn) {
-    p_uty(tags = if (is_required[pn]) c("train", "required") else "train")
+    if (is_required[pn]) {
+      p_uty(tags = c("train", "required"))
+    } else {
+      p_uty(default = "<unknown>", tags = "train")
+    }
   }), param_names)
 
   invoke(ps, .args = args)