Skip to content

Commit 8643394

Browse files
committed
Make the starting colors consistent
1 parent b8fd806 commit 8643394

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lambeq/backend/drawing/drawing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -453,9 +453,9 @@ def _get_box_color(box: grammar.Diagrammable,
453453
if isinstance(box, grammar.Frame) and hasattr(box, 'name'):
454454
frame_attr = getattr(box, f'frame_{coloring_mode}')
455455
if coloring_mode == ColoringMode.TYPE:
456-
frame_attr += (len(FRAME_COLORS) // 7) * box.frame_order
456+
frame_attr += (len(FRAME_COLORS) // 7) * (box.frame_order - 1)
457457

458-
color = FRAME_COLORS[frame_attr % len(FRAME_COLORS)]
458+
color = FRAME_COLORS[(frame_attr - 1) % len(FRAME_COLORS)]
459459

460460
return color
461461

0 commit comments

Comments
 (0)