From eae65ac22b7e6fb4028d97599ffc38e332a3fa25 Mon Sep 17 00:00:00 2001 From: Xiaobo Xia <56142822+xiaoboxia@users.noreply.github.com> Date: Mon, 17 Jul 2023 09:02:10 -0400 Subject: [PATCH] suppress printing data split type (#1126) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * first commit * second update * third update --------- Co-authored-by: “xiaoboxia” <“xiaoboxia.uni@gmail.com”> Co-authored-by: Shaokun Co-authored-by: “skzhang1” <“shaokunzhang529@gmail.com”> --- flaml/automl/automl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py index e37e7abde..7e9e50d08 100644 --- a/flaml/automl/automl.py +++ b/flaml/automl/automl.py @@ -1683,7 +1683,8 @@ class AutoML(BaseEstimator): self._state.fit_kwargs, self._state.groups, ) - logger.info(f"Data split method: {self._split_type}") + if X_val is not None: + logger.info(f"Data split method: {self._split_type}") eval_method = self._decide_eval_method(eval_method, time_budget) self._state.eval_method = eval_method logger.info("Evaluation method: {}".format(eval_method))