Update bubbly.py

AashitaK · May 29, 2018 · 0357efc · 0357efc
1 parent 53ff928
commit 0357efc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bubbly/bubbly.py b/bubbly/bubbly.py
@@ -12,7 +12,7 @@ def bubbleplot(dataset, x_column, y_column, bubble_column, time_column, size_col
     # color_column is not None and categorical, in which case set color_column as None
     category_column = None
     if color_column:
-        if dataset[color_column].dtype.name in ['category', 'object']:
+        if dataset[color_column].dtype.name in ['category', 'object', 'bool']:
             category_column = color_column
             color_column = None