corrected color bug in pie/sunburst etc

Author: emmanuelle

packages/python/plotly/plotly/express/_core.py

@@ -331,7 +331,10 @@ def make_trace_kwargs(args, trace_spec, trace_data, mapping_labels, sizeref):
                         mapping_labels[attr_label] = "%{color}"
                     else:
                         trace_patch["marker"]["colors"] = []
-                        mapping = {}
+                        if args["color_discrete_map"] is not None:
+                            mapping = args["color_discrete_map"].copy()
+                        else:
+                            mapping = {}
                         for cat in trace_data[attr_value]:
                             if mapping.get(cat) is None:
                                 mapping[cat] = args["color_discrete_sequence"][

packages/python/plotly/plotly/tests/test_core/test_px/test_px_functions.py

@@ -218,6 +218,11 @@ def test_sunburst_treemap_with_path_color():
     fig = px.sunburst(df, path=path, color="vendors")
     assert len(np.unique(fig.data[0].marker.colors)) == 9
 
+    # Discrete color and color_discrete_map
+    cmap = {"Tech": "yellow", "Finance": "magenta", "(?)": "black"}
+    fig = px.sunburst(df, path=path, color="sectors", color_discrete_map=cmap)
+    assert np.all(np.in1d(fig.data[0].marker.colors, list(cmap.values())))
+
     # Numerical column in path
     df["regions"] = df["regions"].map({"North": 1, "South": 2})
     path = ["total", "regions", "sectors", "vendors"]