error early if missing DiagrammeR (#8037)

2022-07-02 04:37:53 -07:00 · 2022-07-02 04:37:53 -07:00 · 3af02584c1
commit 3af02584c1
parent bc4f802b17
1 changed files with 3 additions and 0 deletions
--- a/R-package/R/xgb.plot.multi.trees.R
+++ b/R-package/R/xgb.plot.multi.trees.R
@ -62,6 +62,9 @@
 #' @export
 xgb.plot.multi.trees <- function(model, feature_names = NULL, features_keep = 5, plot_width = NULL, plot_height = NULL,
                                 render = TRUE, ...){
+  if (!requireNamespace("DiagrammeR", quietly = TRUE)) {
+    stop("DiagrammeR is required for xgb.plot.multi.trees")
+  }
  check.deprecation(...)
  tree.matrix <- xgb.model.dt.tree(feature_names = feature_names, model = model)