Fix GPU option

bacpop · Dec 18, 2024 · d7f993f · d7f993f
1 parent 78ee2e3
commit d7f993f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PopPUNK/visualise.py b/PopPUNK/visualise.py
@@ -717,7 +717,7 @@ def generate_visualisations(query_db,
                 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:
-                genomeNetwork = remove_nodes_from_graph(genomeNetwork, all_seq, viz_subset, use_gpu = gpu_graph)
+                genomeNetwork = remove_nodes_from_graph(genomeNetwork, all_seq, viz_subset, use_gpu = False)
         elif rank_fit is not None:
             genomeNetwork = sparse_mat_to_network(sparse_mat, combined_seq, use_gpu = gpu_graph)
         elif recalculate_distances: