fix include_unlinked_nodes

srl-labs · Apr 17, 2024 · d5ce7bc · d5ce7bc
1 parent bc2275d
commit d5ce7bc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/clab2drawio.py b/clab2drawio.py
@@ -845,8 +845,10 @@ def main(input_file, output_file, grafana, theme, include_unlinked_nodes=False,
 
     if not include_unlinked_nodes:
         connected_nodes = {name: node for name, node in nodes.items() if node.links}
-    diagram.nodes = connected_nodes
-    nodes = diagram.nodes
+        diagram.nodes = connected_nodes
+        nodes = diagram.nodes
+    else:
+        diagram.nodes = nodes
 
     assign_graphlevels(diagram, verbose=False)
     calculate_positions(diagram, layout=layout, verbose=verbose)