fix(edge_types): add ui config fallback if there is no diect effect

causy-dev · Jul 5, 2024 · 400d1b5 · 400d1b5
1 parent ea1c8db
commit 400d1b5
Showing 1 changed file with 19 additions and 4 deletions.
diff --git a/causy/edge_types.py b/causy/edge_types.py
@@ -29,14 +29,14 @@ class DirectedEdgeUIConfig(EdgeTypeConfig):
     edge_type: str = DirectedEdge().name
 
     default_ui_config: Optional[EdgeUIConfig] = EdgeUIConfig(
-        label_field="direct_effect",
-        color="#0f0fff",
+        label_field="correlation",
+        color="#333",
         width=4,
         style="dashed",
         animated=True,
         marker_start=None,
         marker_end="ArrowClosed",
-        label="Direct Effect: ${direct_effect.toFixed(4)}",
+        label="Direct Effect: ${correlation.toFixed(4)}",
     )
     conditional_ui_configs: Optional[List[ConditionalEdgeUIConfig]] = [
         ConditionalEdgeUIConfig(
@@ -53,7 +53,22 @@ class DirectedEdgeUIConfig(EdgeTypeConfig):
             condition_field="direct_effect",
             condition_value=0,
             condition_comparison=ConditionalEdgeUIConfigComparison.LESS,
-        )
+        ),
+        ConditionalEdgeUIConfig(
+            ui_config=EdgeUIConfig(
+                color="#0f0fff",
+                width=4,
+                style="dashed",
+                animated=True,
+                marker_start=None,
+                marker_end="ArrowClosed",
+                label_field="direct_effect",
+                label="Direct Effect: ${direct_effect.toFixed(4)}",
+            ),
+            condition_field="direct_effect",
+            condition_value=0,
+            condition_comparison=ConditionalEdgeUIConfigComparison.GREATER,
+        ),
     ]