update pipeline for new dfe wit h-s relationship

popsim-consortium · Aug 22, 2023 · 6769863 · 6769863
1 parent eb16df2
commit 6769863
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/workflows/ts2fs.py b/workflows/ts2fs.py
@@ -53,9 +53,10 @@ def f(x):
         mut_types = {}
         for dfe in ts.metadata["stdpopsim"]["DFEs"]:
             for mt in dfe["mutation_types"]:
-                mid = mt["slim_mutation_type_id"]
-                if not mid in mut_types:
-                    mut_types[mid] = "neutral" if mt["is_neutral"] else "non_neutral"
+                mids = mt["slim_mutation_type_id"]
+                for mid in mids:
+                    if not mid in mut_types:
+                        mut_types[mid] = "neutral" if mt["is_neutral"] else "non_neutral"
 
         site_class = np.empty(ts.num_sites, dtype=object)
         for j, s in enumerate(ts.sites()):