diff --git a/tests/testthat/helper-model.R b/tests/testthat/helper-model.R index 181b691..f4dd940 100644 --- a/tests/testthat/helper-model.R +++ b/tests/testthat/helper-model.R @@ -87,7 +87,6 @@ expect_can_tune_boost_tree <- function(model) { mtcars$cyl <- factor(mtcars$cyl) mtcars$vs <- factor(mtcars$vs) - grid_df <- data.frame(trees = c(10, 20)) resamples <- rsample::vfold_cv(mtcars, v = 2) # regression @@ -95,12 +94,12 @@ expect_can_tune_boost_tree <- function(model) { parsnip::set_mode(model, "regression"), mpg ~ ., resamples = resamples, - grid = 3, + grid = 2, metrics = yardstick::metric_set(yardstick::rmse) ) expect_equal(nrow(adj), nrow(resamples)) - expect_equal(nrow(tune::collect_metrics(adj)), 3) + expect_equal(nrow(tune::collect_metrics(adj)), 2) expect_true(all(!is.nan(tune::collect_metrics(adj)$mean))) # classification