Bug fix: do not draw kamada graph when graph is empty

Edinburgh-Genome-Foundry · Dec 4, 2019 · d8f80a3 · d8f80a3
1 parent 806f22b
commit d8f80a3
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/dnacauldron/AssemblyMix/AssemblyMixBase/AssemblyMixPlotsMixin.py b/dnacauldron/AssemblyMix/AssemblyMixBase/AssemblyMixPlotsMixin.py
@@ -147,7 +147,6 @@ def polar(xy):
     def plot_connections_graph(self, ax=None, figsize=(20, 20)):
 
         graph = self.uniquified_connection_graph
-
         def fragment_label(i):
             fragment = self.fragments_dict[i]
             return "\n".join(
@@ -161,13 +160,14 @@ def fragment_label(i):
         labels = {i: fragment_label(i) for i in graph}
         if ax is None:
             fig, ax = plt.subplots(1, 1, figsize=figsize)
-        nx.draw_kamada_kawai(
-            graph,
-            labels=labels,
-            font_color="k",
-            edge_color="grey",
-            font_size=12,
-            node_color="w",
-            node_size=3000,
-        )
+        if len(graph):
+            nx.draw_kamada_kawai(
+                graph,
+                labels=labels,
+                font_color="k",
+                edge_color="grey",
+                font_size=12,
+                node_color="w",
+                node_size=3000,
+            )
         return ax