Skip to content

Commit 8c7d8d4

Browse files
Apply suggestions from code review
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 1a9fe2f commit 8c7d8d4

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

tests/testthat/test_bundle_catboost.R

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,11 @@ test_that("bundling + unbundling catboost fits", {
7171

7272
test_pool <- catboost.load_pool(data = test_data)
7373

74-
catboost.predict(mod_butchered_unbundled, test_pool, prediction_type = "Class")
74+
catboost.predict(
75+
mod_butchered_unbundled,
76+
test_pool,
77+
prediction_type = "Class"
78+
)
7579
},
7680
args = list(
7781
mod_butchered_bundle = mod_butchered_bundle,
@@ -82,7 +86,11 @@ test_that("bundling + unbundling catboost fits", {
8286
# run expectations -----------------------------------------------------------
8387
mod_fit <- fit_model()
8488
test_pool <- catboost::catboost.load_pool(data = iris[, 1:4])
85-
mod_preds <- catboost::catboost.predict(mod_fit, test_pool, prediction_type = "Class")
89+
mod_preds <- catboost::catboost.predict(
90+
mod_fit,
91+
test_pool,
92+
prediction_type = "Class"
93+
)
8694

8795
# check classes
8896
expect_s3_class(mod_bundle, "bundled_catboost.Model")

0 commit comments

Comments
 (0)