No more purple

Author: jessegrabowski

pymc/sampling/parallel.py

@@ -493,7 +493,6 @@ def __init__(
             DivergenceBarColumn(
                 table_column=Column("Progress", ratio=2),
                 diverging_color="tab:red",
-                diverging_finished_color="tab:purple",
                 complete_style=Style.parse("rgb(31,119,180)"),  # tab:blue
                 finished_style=Style.parse("rgb(44,160,44)"),  # tab:green
             ),

pymc/util.py

@@ -671,15 +671,12 @@ def call_column(column, task):
 
 
 class DivergenceBarColumn(BarColumn):
-    def __init__(self, *args, diverging_color="red", diverging_finished_color="purple", **kwargs):
+    def __init__(self, *args, diverging_color="red", **kwargs):
         from matplotlib.colors import to_rgb
 
         self.diverging_color = diverging_color
         self.diverging_rgb = [int(x * 255) for x in to_rgb(self.diverging_color)]
 
-        self.diverging_finished_color = diverging_finished_color
-        self.diverging_finished_rgb = [int(x * 255) for x in to_rgb(self.diverging_finished_color)]
-
         super().__init__(*args, **kwargs)
 
         self.non_diverging_style = self.complete_style
@@ -689,7 +686,7 @@ def callbacks(self, task: "Task"):
         divergences = task.fields.get("divergences", 0)
         if divergences > 0:
             self.complete_style = Style.parse("rgb({},{},{})".format(*self.diverging_rgb))
-            self.finished_style = Style.parse("rgb({},{},{})".format(*self.diverging_finished_rgb))
+            self.finished_style = Style.parse("rgb({},{},{})".format(*self.diverging_rgb))
         else:
             self.complete_style = self.non_diverging_style
             self.finished_style = self.non_diverging_finished_style