diff --git a/tests/testthat/test-catboost.R b/tests/testthat/test-catboost.R index 57dd897..b61c3d6 100644 --- a/tests/testthat/test-catboost.R +++ b/tests/testthat/test-catboost.R @@ -8,7 +8,15 @@ test_that("catboost", { test_that("catboost with tune", { - model <- parsnip::boost_tree(mtry = 5, trees = tune()) + model <- parsnip::boost_tree( + mtry = 5, + learn_rate = tune(), + loss_reduction = tune(), + sample_size = tune(), + trees = tune(), + min_n = tune(), + tree_depth = tune() + ) model <- parsnip::set_engine(model, "catboost") expect_can_tune_boost_tree(model) diff --git a/tests/testthat/test-lightgbm.R b/tests/testthat/test-lightgbm.R index 94c8a02..9fb6678 100644 --- a/tests/testthat/test-lightgbm.R +++ b/tests/testthat/test-lightgbm.R @@ -8,7 +8,15 @@ test_that("lightgbm", { test_that("lightgbm with tune", { - model <- parsnip::boost_tree(mtry = 5, trees = tune()) + model <- parsnip::boost_tree( + mtry = 5, + learn_rate = tune(), + loss_reduction = tune(), + sample_size = tune(), + trees = tune(), + min_n = tune(), + tree_depth = tune() + ) model <- parsnip::set_engine(model, "lightgbm") model <- parsnip::set_mode(model, "regression")