Avoid network modification

bacpop · Nov 15, 2024 · 12e8c8c · 12e8c8c
1 parent 86257c6
commit 12e8c8c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/PopPUNK/plot.py b/PopPUNK/plot.py
@@ -562,16 +562,17 @@ def outputsForCytoscape(G, G_mst, isolate_names, clustering, outPrefix, epiCsv,
     component_assignments, component_hist = gt.label_components(G)
     print("All assignments: " + str(component_assignments))
     for component_idx in range(len(component_hist)):
+        G_copy = G
         remove_list = []
         for vidx, v_component in enumerate(component_assignments.a):
             print("Index: " + str(vidx))
             print("Component: " + str(v_component))
             print("Name: " + str(seqLabels[vidx]))
             if v_component != component_idx:
                 remove_list.append(vidx)
-        G.remove_vertex(remove_list)
-        G.purge_vertices()
-        save_network(G, prefix = outPrefix, suffix = "_component_" + str(component_idx + 1), use_graphml = True)
+        G_copy.remove_vertex(remove_list)
+        G_copy.purge_vertices()
+        save_network(G_copy, prefix = outPrefix, suffix = "_component_" + str(component_idx + 1), use_graphml = True)
 
     if G_mst != None:
         isolate_labels = isolateNameToLabel(G_mst.vp.id)