add argument name which is being overwritten in warning message

microsoft · Nov 3, 2021 · 2d0392e · 2d0392e
1 parent 771cb58
commit 2d0392e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python-package/lightgbm/sklearn.py b/python-package/lightgbm/sklearn.py
@@ -584,7 +584,7 @@ def fit(self, X, y,
         for alias in _ConfigAliases.get('objective'):
             if alias in params:
                 self._objective = params.pop(alias)
-                _log_warning(f"Found `{alias}` in params. Will use it instead of argument")
+                _log_warning(f"Found '{alias}' in params. Will use it instead of 'objective' argument")
         if self._objective is None:
             if isinstance(self, LGBMRegressor):
                 self._objective = "regression"

diff --git a/tests/python_package_test/test_sklearn.py b/tests/python_package_test/test_sklearn.py
@@ -178,7 +178,7 @@ def test_objective_aliases(custom_objective):
     for alias in lgb.basic._ConfigAliases.get('objective'):
         gbm = lgb.LGBMRegressor(n_estimators=5, **{alias: obj})
         if alias != 'objective':
-            with pytest.warns(UserWarning, match=f"Found `{alias}` in params. Will use it instead of argument"):
+            with pytest.warns(UserWarning, match=f"Found '{alias}' in params. Will use it instead of 'objective' argument"):
                 gbm.fit(X_train, y_train, eval_set=[(X_test, y_test)])
         else:
             gbm.fit(X_train, y_train, eval_set=[(X_test, y_test)])