Fix cytoscape network vertex labels

bacpop · Aug 4, 2020 · dcfca15 · dcfca15
1 parent 896159b
commit dcfca15
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/PopPUNK/plot.py b/PopPUNK/plot.py
@@ -394,6 +394,11 @@ def outputsForCytoscape(G, clustering, outPrefix, epiCsv, queryList = None, suff
                 viz_vertex[vertex] = False
         G.set_vertex_filter(viz_vertex)
 
+    # edit names
+    edited_names = isolateNameToLabel(G.vp.id)
+    for n,v in enumerate(G.vertices()):
+        G.vp.id[v] = edited_names[n]
+
     # write graph file
     if suffix is None:
         graph_file_name = os.path.basename(outPrefix) + "_cytoscape.graphml"