fix for #30

kassambara · Apr 4, 2017 · 2c5fe9c · 2c5fe9c
1 parent 802b3a8
commit 2c5fe9c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/NEWS.md b/NEWS.md
@@ -3,6 +3,7 @@
 ## Bug fixes
 
 - Now, the argument `invisible` works properly in the function `fviz_pca_biplot()`([@ginolhac, #26](https://github.com/kassambara/factoextra/issues/26)).
+- The function `fviz_dend()` now works for an object of class `diana` ([@qfazille, #30](https://github.com/kassambara/factoextra/issues/30)).
 
 ## Minor changes
 

diff --git a/R/fviz_dend.R b/R/fviz_dend.R
@@ -131,7 +131,8 @@ fviz_dend <- function(x, k = NULL, h = NULL, k_colors = NULL, palette = NULL,  s
     method <- ""
   }
   else stop("Can't handle an object of class ", paste(class(x), collapse =", ") )
-  if(is.na(method)) method <- ""
+  if(is.null(method)) method <- ""
+  else if(is.na(method)) method <- ""
   if(is.null(sub) & method!="") sub = paste0("Method: ", method)
 
   if(!is.null(dendextend::labels_cex(dend))) cex <- dendextend::labels_cex(dend)