pass call through validate_not_trained

tidymodels · Oct 25, 2024 · cfa2031 · cfa2031
1 parent 72166c4
commit cfa2031
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/R/update.R b/R/update.R
@@ -127,12 +127,13 @@ validate_has_unique_names <- function(x) {
   invisible(x)
 }
 
-validate_not_trained <- function(x) {
+validate_not_trained <- function(x, call = rlang::caller_env()) {
   if (is_trained(x)) {
     step_type <- class(x)[1]
 
     cli::cli_abort(
-      "To update {.fn {step_type}}, it must not be trained."
+      "To update {.fn {step_type}}, it must not be trained.",
+      call = call
     )
   }
 

diff --git a/tests/testthat/_snaps/update.md b/tests/testthat/_snaps/update.md
@@ -11,6 +11,6 @@
     Code
       update(stp, x = 5)
     Condition
-      Error in `validate_not_trained()`:
+      Error in `update()`:
       ! To update `step_stp()`, it must not be trained.