allow prediction column named .pred

simonpcouch · simonpcouch · commit 3385fbce71a7 · 2024-09-18T09:14:24.000-05:00
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

Original file line number	Diff line number	Diff line change
`@@ -114,6 +114,9 @@ predict.numeric_range <- function(object, new_data, tailor, ...) {`
`114`	`114`	`}`
`115`	`115`
`116`	`116`	`rename_prediction_column <- function(data, est_nm) {`
	`117`	`+ if (identical(est_nm, ".pred")) {`
	`118`	`+ return(data)`
	`119`	`+ }`
`117`	`120`	`data[[".pred"]] <- data[[est_nm]]`
`118`	`121`	`data[[est_nm]] <- NULL`
`119`	`122`	`data`