Make the starting colors consistent

CQCL · Nov 6, 2024 · 8643394 · 8643394
1 parent b8fd806
commit 8643394
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lambeq/backend/drawing/drawing.py b/lambeq/backend/drawing/drawing.py
@@ -453,9 +453,9 @@ def _get_box_color(box: grammar.Diagrammable,
         if isinstance(box, grammar.Frame) and hasattr(box, 'name'):
             frame_attr = getattr(box, f'frame_{coloring_mode}')
             if coloring_mode == ColoringMode.TYPE:
-                frame_attr += (len(FRAME_COLORS) // 7) * box.frame_order
+                frame_attr += (len(FRAME_COLORS) // 7) * (box.frame_order - 1)
 
-            color = FRAME_COLORS[frame_attr % len(FRAME_COLORS)]
+            color = FRAME_COLORS[(frame_attr - 1) % len(FRAME_COLORS)]
 
     return color