Fix bug when factor column didn't contain all levels.

jlmelville · Aug 6, 2018 · 99764db · 99764db
1 parent 3a63ace
commit 99764db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/vizier.R b/R/vizier.R
@@ -494,7 +494,7 @@ color_helper_column <- function(x,
 # factor_to_colors(iris$Species, color_scheme = "Set3") # ColorBrewer palette
 # factor_to_colors(iris$Species, color_scheme = rainbow) # color ramp function
 factor_to_colors <- function(x, color_scheme = grDevices::rainbow) {
-  category_names <- unique(x)
+  category_names <- levels(x)
   ncolors <- length(category_names)
   color_palette = make_palette(ncolors = ncolors, color_scheme = color_scheme)
   color_palette[x]