handle ambiguous type when all-custom postprocessing

tidymodels · May 2, 2024 · cd6a2eb · cd6a2eb
1 parent 33ab426
commit cd6a2eb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/R/validation-rules.R b/R/validation-rules.R
@@ -90,6 +90,10 @@ check_duplicates <- function(x, call) {
 }
 
 infer_type <- function(orderings) {
+  if (all(orderings$output_all)) {
+    return("unknown")
+  }
+
   if (all(orderings$output_numeric | orderings$output_all)) {
     return("regression")
   }

diff --git a/tests/testthat/test-validation-rules.R b/tests/testthat/test-validation-rules.R
@@ -105,3 +105,14 @@ test_that("validation of operations (classification)", {
       adjust_probability_threshold(threshold = .4)
   )
 })
+
+test_that("validation of operations (ambiguous type)", {
+  expect_no_condition(
+    ambiguous_ctr <-
+      container() %>%
+      adjust_predictions_custom(squared = .pred^2) %>%
+      adjust_predictions_custom(boop = boop)
+  )
+
+  expect_equal(ambiguous_ctr$type, "unknown")
+})