allow prediction column named .pred

tidymodels · Sep 18, 2024 · 3385fbc · 3385fbc
1 parent 9383acd
commit 3385fbc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/adjust-numeric-range.R b/R/adjust-numeric-range.R
@@ -114,6 +114,9 @@ predict.numeric_range <- function(object, new_data, tailor, ...) {
 }
 
 rename_prediction_column <- function(data, est_nm) {
+  if (identical(est_nm, ".pred")) {
+    return(data)
+  }
   data[[".pred"]] <- data[[est_nm]]
   data[[est_nm]] <- NULL
   data