Fix selection of between strain cluster

bacpop · Oct 8, 2024 · f166cfa · f166cfa
1 parent 4d8f1b2
commit f166cfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PopPUNK/bgmm.py b/PopPUNK/bgmm.py
@@ -66,7 +66,7 @@ def findBetweenLabel_bgmm(means, assignments):
         most_dists[mixture_component] = np.count_nonzero(assignments == mixture_component)
 
     sorted_dists = sorted(most_dists.items(), key=operator.itemgetter(1), reverse=True)
-    return(sorted_dists[rank][0])
+    return(sorted_dists[0][0])
 
 def findWithinLabel(means, assignments, rank = 0):
     """Identify within-strain links