fixed orientation of hint and requirement nodes in generate_graph

lanl · Sep 12, 2024 · d65a90f · d65a90f
1 parent c2f081a
commit d65a90f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/beeflow/common/gdb/generate_graph.py b/beeflow/common/gdb/generate_graph.py
@@ -72,7 +72,8 @@ def add_edges_to_dot(graph, dot):
     """Add edges from the graph to the Graphviz object with appropriate labels."""
     for source, target, attributes in graph.edges(data=True):
         edge_label = attributes.get('label', '')
-        if edge_label in ('INPUT_OF', 'DESCRIBES'):
+        if edge_label in ('INPUT_OF', 'DESCRIBES',
+                          'HINT_OF', 'REQUIREMENT_OF'):
             dot.edge(source, target, label=edge_label, fontsize="10")
         else:
             dot.edge(target, source, label=edge_label, fontsize="10")