Revert node labelling

bacpop · Nov 15, 2024 · 575b87b · 575b87b
1 parent a372e4d
commit 575b87b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/PopPUNK/visualise.py b/PopPUNK/visualise.py
@@ -699,7 +699,7 @@ def generate_visualisations(query_db,
             if gpu_graph:
                 genomeNetwork = cugraph_to_graph_tool(genomeNetwork, isolateNameToLabel(all_seq))
             # Hard delete from network to remove samples (mask doesn't work neatly)
-            if include_files is not None and not use_partial_query_graph:
+            if include_files is not None:
                 genomeNetwork = remove_nodes_from_graph(genomeNetwork, all_seq, viz_subset, use_gpu = gpu_graph)
         elif rank_fit is not None:
             genomeNetwork = sparse_mat_to_network(sparse_mat, combined_seq, use_gpu = gpu_graph)
@@ -708,8 +708,7 @@ def generate_visualisations(query_db,
             sys.exit(1)
         # If network has been pruned then only use the appropriate subset of names - otherwise use all names
         # for full network
-        node_labels = viz_subset if (use_partial_query_graph is not None or include_files is not None) \
-                                      else combined_seq
+        node_labels = viz_subset if use_partial_query_graph is not None else combined_seq
         sys.stderr.write('Preparing outputs for cytoscape\n')
         outputsForCytoscape(genomeNetwork,
                             mst_graph,