Make set colors an attribute of SetDiagram (issue #5)

paulbrodersen · Oct 16, 2024 · 6d59c2d · 6d59c2d
1 parent dc1c628
commit 6d59c2d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/matplotlib_set_diagrams/_diagram_classes.py b/matplotlib_set_diagrams/_diagram_classes.py
@@ -98,13 +98,14 @@ def __init__(
         if set_colors is None:
             set_colors = plt.rcParamsDefault['axes.prop_cycle'].by_key()['color']
             set_colors = set_colors[:total_sets]
-        self.subset_colors = self._get_subset_colors(subset_ids, set_colors)
+        self.set_colors = set_colors
+        self.subset_colors = self._get_subset_colors(subset_ids, self.set_colors)
 
         self.ax = self._initialize_axis(ax=ax)
         self.subset_artists = self._draw_subsets(
             self.subset_geometries, self.subset_colors, self.ax)
         self.set_artists = self._draw_sets(
-            origins, radii, set_colors, self.ax)
+            origins, radii, self.set_colors, self.ax)
 
         if subset_labels:
             self.subset_label_artists = self._draw_subset_labels(