Tune schedule

NAMESPACE

@@ -204,13 +204,15 @@ export(fit_resamples)
 export(forge_from_workflow)
 export(get_metric_time)
 export(get_tune_colors)
+export(get_tune_schedule)
 export(initialize_catalog)
 export(int_pctl)
 export(is_preprocessor)
 export(is_recipe)
 export(is_workflow)
 export(last_fit)
 export(load_pkgs)
+export(loopy)
 export(maybe_choose_eval_time)
 export(message_wrap)
 export(metrics_info)

R/0_imports.R

@@ -47,7 +47,7 @@ utils::globalVariables(
     "rowwise", ".best", "location", "msg", "..object", ".eval_time",
     ".pred_survival", ".pred_time", ".weight_censored", "nice_time",
     "time_metric", ".lower", ".upper", "i", "results", "term", ".alpha",
-    ".method", "old_term", ".lab_pre", ".model", ".num_models"
+    ".method", "old_term", ".lab_pre", ".model", ".num_models", "predict_stage"
   )
 )
 

R/grid_helpers.R

@@ -151,6 +151,11 @@
 }
 
 predict_wrapper <- function(model, new_data, type, eval_time, subgrid = NULL) {
+  if (length(type) > 1) {
+    cli::cli_abort("Argument {.arg type} in {.fn predict_wrapper} should be
+                   a single value, not {.obj_type_friendly {type}}.",
+                   call = NULL)
+  }
   if (is.null(subgrid)) {
     fn <- "predict.model_fit"
   } else {