Fix for n=1 case with create_using

networkx · Nov 15, 2020 · 39fd7d5 · 39fd7d5
1 parent d898943
commit 39fd7d5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/networkx/generators/trees.py b/networkx/generators/trees.py
@@ -218,10 +218,10 @@ def random_tree(n, seed=None, create_using=None):
         raise nx.NetworkXPointlessConcept("the null graph is not a tree")
     # Cannot create a Prüfer sequence unless `n` is at least two.
     if n == 1:
-        return nx.empty_graph(1)
-
-    sequence = [seed.choice(range(n)) for i in range(n - 2)]
-    utree = nx.from_prufer_sequence(sequence)
+        utree = nx.empty_graph(1)
+    else:
+        sequence = [seed.choice(range(n)) for i in range(n - 2)]
+        utree = nx.from_prufer_sequence(sequence)
 
     if create_using is None:
         tree = utree