bugfix: node labels not initialized with all node values

IntelLabs · Dec 26, 2022 · 44e348b · 44e348b
1 parent 6ee347c
commit 44e348b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/plot_utils/draw_graph.py b/plot_utils/draw_graph.py
@@ -176,6 +176,11 @@ def draw_graph(graph, latent_nodes=None, selection_nodes=None, bkcolor='white',
         selection_nodes = set()
     if latent_nodes is None:
         latent_nodes = set()
+    if node_labels is None:
+        node_labels = {}
+    for node in graph.nodes_set:
+        if node not in node_labels:
+            node_labels[node] = node
 
     bottom = 0
     # top = 1