Update for python 3.11

davidemms · Oct 7, 2024 · 92d520d · 92d520d
1 parent 1ee31c0
commit 92d520d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts_of/accelerate.py b/scripts_of/accelerate.py
@@ -357,7 +357,7 @@ def sample_random(og, n_max):
     Returns:
         genes - list of genes
     """
-    return random.sample(og, min(n_max, len(og)))
+    return random.sample(sorted(og), min(n_max, len(og)))
 
 
 def get_new_species_clades(rooted_species_tree_fn, core_species_ids, n_core_species=2):

diff --git a/scripts_of/sample_genes.py b/scripts_of/sample_genes.py
@@ -260,7 +260,7 @@ def select_from_aligned(infn, n_sample, q_trim=True):
     if n_extra > 0:
         # select some more to make it up to the total 
         not_used = set(range(n_keep)).difference(cluster_representative)
-        cluster_representative.extend(random.sample(not_used, n_extra))
+        cluster_representative.extend(random.sample(sorted(not_used), n_extra))
         # print("Found extra sequences. Have %d" % len(cluster_representative))
     # print(cluster_representative)
     selected = [accs[d_new_old[i]] for i in cluster_representative]