DEPENDS_ON arrows and text bold (#930)

* Make DEPENDS_ON and RESTARTED_FROM arrows and text bold for dag --------- Co-authored-by: leahh <[email protected]>
lanl · Sep 24, 2024 · cd51dc2 · cd51dc2
1 parent f52049c
commit cd51dc2
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/beeflow/common/gdb/generate_graph.py b/beeflow/common/gdb/generate_graph.py
@@ -55,7 +55,7 @@ def get_node_label_and_color(label, attributes, label_to_color):
     """Return the appropriate node label and color based on node type."""
     label_to_attribute = {
         ":Workflow": "Workflow",
-        ":Output": attributes.get('value', label),
+        ":Output": attributes.get('glob', label),
         ":Metadata": attributes.get('state', label),
         ":Task": attributes.get('name', label),
         ":Input": attributes.get('source', label),
@@ -77,5 +77,8 @@ def add_edges_to_dot(graph, dot):
         edge_label = attributes.get('label', '')
         if edge_label in ('INPUT_OF', 'DESCRIBES', 'HINT_OF', 'REQUIREMENT_OF'):
             dot.edge(source, target, label=edge_label, fontsize="10")
+        elif edge_label in ('DEPENDS_ON', 'RESTARTED_FROM'):
+            dot.edge(target, source, label=edge_label, penwidth="3",
+                     fontsize="10", fontname="times-bold")
         else:
             dot.edge(target, source, label=edge_label, fontsize="10")
diff --git a/docs/sphinx/images/941cd5.png b/docs/sphinx/images/941cd5.png
diff --git a/docs/sphinx/images/cat-dag.png b/docs/sphinx/images/cat-dag.png
diff --git a/docs/sphinx/visualization.rst b/docs/sphinx/visualization.rst
@@ -22,7 +22,7 @@ Example DAG
 The DAG below was created by running the dag command while the cat-grep-tar
 example workflow was running.
 
-.. image:: images/941cd5.png
+.. image:: images/cat-dag.png
 
 The orange bubbles are inputs, the blue bubbles are task states, the red
 bubbles are tasks, and the green bubbles are outputs. The graph is in a