From ee1daee45ab9a99aeebd18c9f6dc341aece85e49 Mon Sep 17 00:00:00 2001 From: 34j <55338215+34j@users.noreply.github.com> Date: Fri, 5 May 2023 22:24:23 +0900 Subject: [PATCH] fix(callback): do not use early stopping in `test_progress_bar_binary()` --- tests/python_package_test/test_callback.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python_package_test/test_callback.py b/tests/python_package_test/test_callback.py index 11050fcc4303..8d4c718a0ef2 100644 --- a/tests/python_package_test/test_callback.py +++ b/tests/python_package_test/test_callback.py @@ -81,7 +81,7 @@ def test_progress_bar_binary(): X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.1, random_state=42) gbm = lgb.LGBMClassifier(n_estimators=50, verbose=-1) callback = lgb.progress_bar() - gbm.fit(X_train, y_train, eval_set=[(X_test, y_test)], callbacks=[lgb.early_stopping(5), callback]) + gbm.fit(X_train, y_train, eval_set=[(X_test, y_test)], callbacks=[callback]) assert issubclass(callback.tqdm_cls, tqdm.std.tqdm) assert isinstance(callback.pbar, tqdm.std.tqdm)