Add crate to custom_check in PipeOpTuneThreshold.R

mlr-org · Aug 8, 2024 · c6272ce · c6272ce
1 parent 1859076
commit c6272ce
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/R/PipeOpTuneThreshold.R b/R/PipeOpTuneThreshold.R
@@ -84,8 +84,10 @@ PipeOpTuneThreshold = R6Class("PipeOpTuneThreshold",
       ps = ps(
         measure = p_uty(custom_check = check_class_or_character("Measure", mlr_measures), tags = "train"),
         optimizer = p_uty(custom_check = check_optimizer, tags = "train"),
-        log_level = p_uty(tags = "train",
-          function(x) check_string(x) %check||% check_integerish(x))
+        log_level = p_uty(
+          custom_check = crate(function(x) check_string(x) %check||% check_integerish(x), .parent = topenv()),
+          tags = "train"
+        )
       )
       ps$values = list(measure = "classif.ce", optimizer = "gensa", log_level = "warn")
       super$initialize(id, param_set = ps, param_vals = param_vals, packages = "bbotk",