moved check to function; small mesage edits

Author: topepo

R/linear_reg.R

@@ -112,13 +112,7 @@ translate.linear_reg <- function(x, engine = x$engine, ...) {
     # Since the `fit` information is gone for the penalty, we need to have an
     # evaluated value for the parameter.
     x$args$penalty <- rlang::eval_tidy(x$args$penalty)
-    if (length(x$args$penalty) != 1) {
-      rlang::abort(c(
-        "For the glmnet engine, `penalty` must be a single number.",
-        glue::glue("There are {length(x$args$penalty)} values for `penalty`."),
-        "To try multiple values for total regularization, use the tune package."
-      ))
-    }
+    check_glmnet_penalty(x)
   }
 
   x

R/logistic_reg.R

@@ -115,13 +115,7 @@ translate.logistic_reg <- function(x, engine = x$engine, ...) {
     # Since the `fit` information is gone for the penalty, we need to have an
     # evaluated value for the parameter.
     x$args$penalty <- rlang::eval_tidy(x$args$penalty)
-    if (length(x$args$penalty) != 1) {
-      rlang::abort(c(
-        "For the glmnet engine, `penalty` must be a single number.",
-        glue::glue("There are {length(x$args$penalty)} values for `penalty`."),
-        "To try multiple values for total regularization, use the tune package."
-      ))
-    }
+    check_glmnet_penalty(x)
   }
 
   if (engine == "LiblineaR") {

R/misc.R

@@ -323,4 +323,13 @@ stan_conf_int <- function(object, newdata) {
   rlang::eval_tidy(fn)
 }
 
-
+check_glmnet_penalty <- function(x) {
+  if (length(x$args$penalty) != 1) {
+    rlang::abort(c(
+      "For the glmnet engine, `penalty` must be a single number (or a value of `tune()`).",
+      glue::glue("There are {length(x$args$penalty)} values for `penalty`."),
+      "To try multiple values for total regularization, use the tune package.",
+      "To predict multiple penalties, use `multi_predict()`"
+    ))
+  }
+}