Fixed bug for colors and color_cycles

Chilipp · Jul 7, 2018 · fc57da8 · fc57da8
1 parent e3924d8
commit fc57da8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/psy_simple/plotters.py b/psy_simple/plotters.py
@@ -1610,7 +1610,14 @@ def update(self, value):
                     np.linspace(0., 1., len(list(self.iter_data)),
                                 endpoint=True)))
             except (ValueError, TypeError, KeyError):
-                self.color_cycle = cycle(slist(value))
+                try:
+                    # do not use safe_list, because it might be a generator
+                    validate_color(value)
+                except (ValueError, TypeError, AttributeError):
+                    pass
+                else:
+                    value = [value]
+                self.color_cycle = cycle(iter(value))
         if changed:
             self.colors = [
                 next(self.color_cycle) for arr in self.iter_data]