[python] Improve error message for plot_metric with Booster (#4709)

* Improve error message for plot_metric with Booster * Update error message
microsoft · Oct 26, 2021 · 1cc84b3 · 1cc84b3
1 parent f497068
commit 1cc84b3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python-package/lightgbm/plotting.py b/python-package/lightgbm/plotting.py
@@ -342,6 +342,8 @@ def plot_metric(
         eval_results = deepcopy(booster.evals_result_)
     elif isinstance(booster, dict):
         eval_results = deepcopy(booster)
+    elif isinstance(booster, Booster):
+        raise TypeError("booster must be dict or LGBMModel. To use plot_metric with Booster type, first record the metrics using record_evaluation callback then pass that to plot_metric as argument `booster`")
     else:
         raise TypeError('booster must be dict or LGBMModel.')