Update spring layout

oscarhiggott · Apr 2, 2021 · 4bfaee8 · 4bfaee8
1 parent 42e8611
commit 4bfaee8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pymatching/matching.py b/src/pymatching/matching.py
@@ -374,7 +374,7 @@ def draw(self):
         warnings.filterwarnings("ignore",category=matplotlib.cbook.mplDeprecation)
         warnings.filterwarnings("ignore", category=DeprecationWarning)
         G = self.to_networkx()
-        pos=nx.spring_layout(G, weight=1)
+        pos=nx.spring_layout(G, weight=None)
         c = "#bfbfbf"
         ncolors = ['w' if n[1]['is_boundary'] else c for n in G.nodes(data=True)]
         nx.draw_networkx_nodes(G, pos=pos, node_color=ncolors, edgecolors=c)