make error context more informative for ranger prediction (closes #1200)

simonpcouch · simonpcouch · commit 6bf39b57a770 · 2024-09-10T15:12:07.000-04:00
diff --git a/R/rand_forest_data.R b/R/rand_forest_data.R
@@ -206,7 +206,8 @@ set_pred(
           c(
             "`ranger` model does not appear to use class probabilities.",
             "i" = "Was the model fit with `probability = TRUE`?"
-          )
+          ),
+          call = call2("predict")
         )
       x
     },
diff --git a/tests/testthat/_snaps/rand_forest_ranger.md b/tests/testthat/_snaps/rand_forest_ranger.md
@@ -20,7 +20,7 @@
       parsnip:::predict_classprob.model_fit(no_prob_model, new_data = lending_club[1:
       6, num_pred])
     Condition
-      Error in `object$spec$method$pred$prob$pre()`:
+      Error in `predict()`:
       ! `ranger` model does not appear to use class probabilities.
       i Was the model fit with `probability = TRUE`?
 

Original file line number	Diff line number	Diff line change
`@@ -206,7 +206,8 @@ set_pred(`
`206`	`206`	`c(`
`207`	`207`	"`ranger` model does not appear to use class probabilities.",
`208`	`208`	"i" = "Was the model fit with `probability = TRUE`?"
`209`		`- )`
	`209`	`+ ),`
	`210`	`+ call = call2("predict")`
`210`	`211`	`)`
`211`	`212`	`x`
`212`	`213`	`},`