From 44235fdf127283e7c846ec292e7c8acbb93fad68 Mon Sep 17 00:00:00 2001
From: Vitalie Spinu <spinuvit@gmail.com>
Date: Sat, 27 Aug 2022 22:27:07 +0200
Subject: [PATCH] Make .generateYColor robust towards NaN and degenerate trees

---
 R/visTree.R | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/R/visTree.R b/R/visTree.R
index 7e8fb3a..b34775c 100644
--- a/R/visTree.R
+++ b/R/visTree.R
@@ -779,9 +779,9 @@ subsetRpart <- function(tree,data,  node = 1L) {
     vardecidedClust <- round(object$frame$yval, digits)
     
     # palette
-    if(length(row.names(object$frame)) > 1){
-      meanV <- object$frame$yval-min(object$frame$yval)
-      meanV <- meanV/max(meanV)
+    if (length(unique(na.omit(object$frame$yval))) > 1) {
+      meanV <- object$frame$yval - min(object$frame$yval, na.rm = TRUE)
+      meanV <- meanV/max(meanV, na.rm = TRUE)
     } else {
       meanV <- 1
     }
@@ -960,4 +960,4 @@ visTreeEditor <- function(data, ...){
          '<script type="text/javascript">',
          ttr,
          '</script>')
-}
\ No newline at end of file
+}