Fix edge assignment by model

bacpop · Dec 20, 2024 · 5bbe6af · 5bbe6af
1 parent 668c1af
commit 5bbe6af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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: