From 5bbe6af4aca655ae0e5a857f54ee5be550aa7246 Mon Sep 17 00:00:00 2001 From: Nick Croucher Date: Fri, 20 Dec 2024 09:23:29 +0000 Subject: [PATCH] Fix edge assignment by model --- PopPUNK/network.py | 2 +- PopPUNK/visualise.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/PopPUNK/network.py b/PopPUNK/network.py index 49d95dd1..f64130c8 100644 --- a/PopPUNK/network.py +++ b/PopPUNK/network.py @@ -2031,7 +2031,7 @@ def generate_network_from_distances(mode = None, threads = 1, gpu_graph = False): """ - Removes all components that do not contain a query sequence. + Generates a network from a distance matrix. Args: mode (str) diff --git a/PopPUNK/visualise.py b/PopPUNK/visualise.py index 96a6bdf5..8ee75ad6 100644 --- a/PopPUNK/visualise.py +++ b/PopPUNK/visualise.py @@ -602,6 +602,7 @@ def generate_visualisations(query_db, acc_distMat = acc_distMat, combined_seq = combined_seq, distance_type = mst_distances, + model = model, threads = threads, gpu_graph = gpu_graph) elif use_sparse: @@ -611,6 +612,7 @@ def generate_visualisations(query_db, rlist = rlist, old_rlist = old_rlist, distance_type = mst_distances, + model = model, threads = threads, gpu_graph = gpu_graph) else: