Skip to content

Commit 0f1b45f

Browse files
committed
Fix Spark tests for tree models
1 parent e64d1a7 commit 0f1b45f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/testthat/test_boost_tree_spark.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ test_that('spark execution', {
3030
boost_tree(trees = 5, mode = "regression") %>%
3131
set_engine("spark", seed = 12),
3232
control = ctrl,
33-
class ~ .,
33+
compounds ~ .,
3434
data = hpc_bt_tr
3535
),
3636
regexp = NA

tests/testthat/test_rand_forest_spark.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ test_that('spark execution', {
3131
rand_forest(trees = 5, mode = "regression") %>%
3232
set_engine("spark", seed = 12),
3333
control = ctrl,
34-
Sepal_Length ~ .,
34+
compounds ~ .,
3535
data = hpc_rf_tr
3636
),
3737
regexp = NA
@@ -44,7 +44,7 @@ test_that('spark execution', {
4444
rand_forest(trees = 5, mode = "regression") %>%
4545
set_engine("spark", seed = 12),
4646
control = ctrl,
47-
Sepal_Length ~ .,
47+
compounds ~ .,
4848
data = hpc_rf_tr
4949
),
5050
regexp = NA

0 commit comments

Comments
 (0)