From da2ce51446f57b66a1ebff08b66e522b3cb8d40e Mon Sep 17 00:00:00 2001 From: Robin Leroy Date: Thu, 2 Jan 2025 03:06:14 +0100 Subject: [PATCH] white on blue --- osaca/semantics/kernel_dg.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/osaca/semantics/kernel_dg.py b/osaca/semantics/kernel_dg.py index 416ecb7..2dd46fb 100644 --- a/osaca/semantics/kernel_dg.py +++ b/osaca/semantics/kernel_dg.py @@ -599,7 +599,10 @@ class KernelDG(nx.DiGraph): else: graph.nodes[n]["style"] += ",filled" graph.nodes[n]["fillcolor"] = color - if (max_color >= 4 and color == 1) or (max_color >= 10 and color in (1, 2, max_color)): + if ( + (max_color >= 4 and color in (1, max_color)) or + (max_color >= 10 and color in (1, 2, max_color - 1 , max_color)) + ): graph.nodes[n]["fontcolor"] = "white" for (u, v), color in edge_colors.items(): # The backward edge of the cycle is represented as the corresponding forward