Merge pull request #311 from gyorilab/comparison-colors

Comparison colors
gyorilab · Mar 23, 2024 · 75da2b4 · 75da2b4
2 parents 4352191 + ab50ba4
commit 75da2b4
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/mira/metamodel/comparison.py b/mira/metamodel/comparison.py
@@ -19,6 +19,11 @@
 from .utils import safe_parse_expr
 
 
+TAG1_COLOR = "blue"
+TAG2_COLOR = "green"
+MERGE_COLOR = "orange"
+
+
 class DataNode(BaseModel):
     """A node in a ModelComparisonGraphdata"""
 
@@ -409,9 +414,9 @@ def __init__(
         refinement_function: Callable[[str, str], bool],
         tag1: str = "1",
         tag2: str = "2",
-        tag1_color: str = "orange",
-        tag2_color: str = "blue",
-        merge_color: str = "red",
+        tag1_color: str = TAG1_COLOR,
+        tag2_color: str = TAG2_COLOR,
+        merge_color: str = MERGE_COLOR,
     ):
         """Create a TemplateModelDelta
 
@@ -428,11 +433,11 @@ def __init__(
         tag2 :
             The tag for the second template model. Default: "2"
         tag1_color :
-            The color for the first template model. Default: "orange"
+            The color for the first template model. Default: "blue"
         tag2_color :
-            The color for the second template model. Default: "blue"
+            The color for the second template model. Default: "green"
         merge_color :
-            The color for the merged template model. Default: "red"
+            The color for the merged template model. Default: "orange"
         """
         self.refinement_func = refinement_function
         self.template_model1 = template_model1
@@ -681,9 +686,9 @@ def for_jupyter(
             name="model.png",
             tag1="1",
             tag2="2",
-            tag1_color="blue",
-            tag2_color="green",
-            merge_color="orange",
+            tag1_color=TAG1_COLOR,
+            tag2_color=TAG2_COLOR,
+            merge_color=MERGE_COLOR,
             prog: str = "dot",
             args: str = "",
             format: Optional[str] = None,