From b6bb3c93da3c28dfbdaf11748f6b03015a991155 Mon Sep 17 00:00:00 2001 From: RektPunk Date: Sun, 15 Dec 2024 13:41:41 +0900 Subject: [PATCH] remove num tree parameter --- tests/python_package_test/test_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python_package_test/test_engine.py b/tests/python_package_test/test_engine.py index 226727ad6985..2f6c0e6befd9 100644 --- a/tests/python_package_test/test_engine.py +++ b/tests/python_package_test/test_engine.py @@ -2310,7 +2310,7 @@ def test_refit(): def test_refit_with_one_tree(): X, y = load_breast_cancer(return_X_y=True) lgb_train = lgb.Dataset(X, label=y) - params = {"objective": "binary", "num_trees": 1, "verbosity": -1} + params = {"objective": "binary", "verbosity": -1} model = lgb.train(params, lgb_train, num_boost_round=1) model_refit = model.refit(X, y) assert isinstance(model_refit, lgb.Booster)