Merge pull request #1058 from tidymodels/augment-message

make missing eval_time error in augment more informative
tidymodels · Jan 24, 2024 · 7c40966 · 7c40966
2 parents 5ba98ee + b04e0e7
commit 7c40966
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: parsnip
 Title: A Common API to Modeling and Analysis Functions
-Version: 1.1.1.9007
+Version: 1.1.1.9008
 Authors@R: c(
     person("Max", "Kuhn", , "[email protected]", role = c("aut", "cre")),
     person("Davis", "Vaughan", , "[email protected]", role = "aut"),

diff --git a/R/augment.R b/R/augment.R
@@ -128,8 +128,12 @@ augment_censored <- function(x, new_data, eval_time = NULL) {
 
   if (spec_has_pred_type(x, "survival")) {
     if (is.null(eval_time)) {
-      rlang::abort(
-        "The `eval_time` argument is missing, with no default.",
+      cli::cli_abort(
+        c(
+          x = "The {.arg eval_time} argument is missing, with no default.",
+          i = "{.arg eval_time} is required to be able to calculate \\
+              predictions of survival probability."
+        ),
         call = caller_env()
       )
     }