def test_run_show_progress_bar(self, show_progress_bar: bool, expected: int) -> None:
params: Dict = {"verbose": -1}
dataset = lgb.Dataset(np.zeros((10, 10)))
study = optuna.create_study()
tuner = LightGBMTunerCV(
params, dataset, study=study, time_budget=1, show_progress_bar=show_progress_bar
)
with mock.patch.object(
_OptunaObjectiveCV, "_get_cv_scores", return_value=[1.0]
), mock.patch("tqdm.tqdm") as mock_tqdm:
tuner.run()
assert mock_tqdm.call_count == expected
def test_run_verbosity(self, verbosity: int, level: int) -> None:
# We need to reconstruct our default handler to properly capture stderr.
optuna.logging._reset_library_root_logger()
optuna.logging.set_verbosity(optuna.logging.INFO)
params: Dict = {"verbose": -1}
dataset = lgb.Dataset(np.zeros((10, 10)))
study = optuna.create_study()
with warnings.catch_warnings():
warnings.simplefilter("ignore", category=FutureWarning)
tuner = LightGBMTunerCV(
params, dataset, study=study, verbosity=verbosity, time_budget=1
)
with mock.patch.object(_OptunaObjectiveCV, "_get_cv_scores", return_value=[1.0]):
tuner.run()
assert optuna.logging.get_verbosity() == level
assert tuner.lgbm_params["verbose"] == -1
