Show colors at the top of the colorbar when creating a gantt with custom colors

Author: omridanan

plotly/figure_factory/_gantt.py

@@ -483,6 +483,24 @@ def gantt_dict(chart, colors, title, index_col, show_colorbar, bar_width,
     if group_tasks:
         task_names.reverse()
 
+    if show_colorbar is True:
+        # generate dummy data to generate legend
+        showlegend = True
+        for k, index_value in enumerate(index_vals):
+            data.append(
+                dict(
+                    x=[tasks[index]['x0'], tasks[index]['x0']],
+                    y=[k, k],
+                    showlegend=True,
+                    hoverinfo='none',
+                    name=str(index_value),
+                    marker=dict(
+                        color=colors[index_value],
+                        size=1
+                    )
+                )
+            )
+
     for index in range(len(tasks)):
         tn = tasks[index]['name']
         del tasks[index]['name']
@@ -510,24 +528,6 @@ def gantt_dict(chart, colors, title, index_col, show_colorbar, bar_width,
             del tasks[index]['description']
         data.append(entry)
 
-    if show_colorbar is True:
-        # generate dummy data to generate legend
-        showlegend = True
-        for k, index_value in enumerate(index_vals):
-            data.append(
-                dict(
-                    x=[tasks[index]['x0'], tasks[index]['x0']],
-                    y=[k, k],
-                    showlegend=True,
-                    hoverinfo='none',
-                    name=str(index_value),
-                    marker=dict(
-                        color=colors[index_value],
-                        size=1
-                    )
-                )
-            )
-
     layout = dict(
         title=title,
         showlegend=showlegend,