Fix error when requesting more than 10 colors

automl · Oct 27, 2023 · f3e7c5a · f3e7c5a
1 parent fea1a2a
commit f3e7c5a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,7 @@
 
 ## Bug-Fixes
 - Fix seaborn style name (#82).
+- Fix error when requesting more than 10 colors in a plot (36 colors available now).
 
 # Version 1.1.2
 

diff --git a/deepcave/utils/styled_plotty.py b/deepcave/utils/styled_plotty.py
@@ -69,9 +69,12 @@ def hex_to_rgb(hex_string: str) -> Tuple[int, int, int]:
 
 def get_color(id_: int, alpha: float = 1) -> Union[str, Tuple[float, float, float, float]]:
     """
-    Currently (Plotly version 5.3.1) there are 10 possible colors.
+    Using Plotly palette for the first 10 ids and Alphabet palette for the next 26, currently 36 colors are possible.
     """
-    color = px.colors.qualitative.Plotly[id_]
+    if id_ < 10:
+        color = px.colors.qualitative.Plotly[id_]
+    else:
+        color = px.colors.qualitative.Alphabet[id_ % 10]
 
     r, g, b = hex_to_rgb(color)
     return f"rgba({r}, {g}, {b}, {alpha})"