format tree handles undirected edges better

rocky · rocky · commit 2b374c4a9ff6 · 2020-12-04T23:08:17.000-05:00
diff --git a/mathicsscript/format.py b/mathicsscript/format.py
@@ -119,7 +119,7 @@ def hierarchy_pos(
 
     # These get swapped if tree edge directions point to the root.
     decendants = nx.descendants
-    out_degree = G.out_degree
+    out_degree = G.out_degree if hasattr(G, "out_degree") else G.degree
     neighbors = G.neighbors
 
     if root is None:
