From 5e941269639a9b9a3322d412ceb46887f540107e Mon Sep 17 00:00:00 2001
From: El Potaeto <pommedeterresautee@msn.com>
Date: Sun, 25 Jan 2015 21:07:06 +0100
Subject: [PATCH] fix mermaid

---
 R-package/DESCRIPTION       | 2 +-
 R-package/NAMESPACE         | 2 +-
 R-package/R/xgb.plot.tree.R | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/R-package/DESCRIPTION b/R-package/DESCRIPTION
index 63ed9581c..26fc3c8ae 100644
--- a/R-package/DESCRIPTION
+++ b/R-package/DESCRIPTION
@@ -25,4 +25,4 @@ Imports:
     data.table (>= 1.9.4),
     magrittr (>= 1.5),
     stringr,
-    DiagrammeR
+    DiagrammeR (>=0.3)
diff --git a/R-package/NAMESPACE b/R-package/NAMESPACE
index d29ad7a18..92485b9c7 100644
--- a/R-package/NAMESPACE
+++ b/R-package/NAMESPACE
@@ -18,7 +18,7 @@ exportMethods(predict)
 import(methods)
 importClassesFrom(Matrix,dgCMatrix)
 importClassesFrom(Matrix,dgeMatrix)
-importFrom(DiagrammeR,DiagrammeR)
+importFrom(DiagrammeR,mermaid)
 importFrom(data.table,":=")
 importFrom(data.table,as.data.table)
 importFrom(data.table,copy)
diff --git a/R-package/R/xgb.plot.tree.R b/R-package/R/xgb.plot.tree.R
index 662ccb21b..0747f0a14 100644
--- a/R-package/R/xgb.plot.tree.R
+++ b/R-package/R/xgb.plot.tree.R
@@ -15,7 +15,7 @@
 #' @importFrom stringr str_split
 #' @importFrom stringr str_extract
 #' @importFrom stringr str_trim
-#' @importFrom DiagrammeR DiagrammeR
+#' @importFrom DiagrammeR mermaid
 #' @param feature_names names of each feature as a character vector. Can be extracted from a sparse matrix (see example). If model dump already contains feature names, this argument should be \code{NULL}.
 #' @param filename_dump the path to the text file storing the model. Model dump must include the gain per feature and per tree (parameter \code{with.stats = T} in function \code{xgb.dump}). Possible to provide a model directly (see \code{model} argument).
 #' @param model generated by the \code{xgb.train} function. Avoid the creation of a dump file.
@@ -85,7 +85,7 @@ xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NU
   no <- allTrees[Feature!="Leaf", c(No)] %>% paste(collapse = ",") %>% paste("class ", ., " redNode", sep = "")
   
   path <- allTrees[Feature!="Leaf", c(yesPath, noPath)] %>% .[order(.)] %>% paste(sep = "", collapse = ";") %>% paste("graph LR", .,collapse = "", sep = ";") %>% paste(CSSstyle, yes, no, sep = ";")
-  DiagrammeR(path, width, height)
+  mermaid(path, width, height)
 }
 
 # Avoid error messages during CRAN check.