Fixed datatype mismatch in multiple_alleles func

ssi-dk · Dec 12, 2024 · 8e8a5e7 · 8e8a5e7
1 parent f02900f
commit 8e8a5e7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/bifrost_chewbbaca/datadump.py b/bifrost_chewbbaca/datadump.py
@@ -21,9 +21,8 @@ def call_percent(calls):
 
 
 def multiple_alleles(calls):
-    return round(
-        len([x for x in calls if x.startswith("NIPH") or x.startswith("NIPHEM")])
-    )
+    multi_allele_codes = set("NIPH","NIPHEM")
+    return len([x for x in calls if x in multi_allele_codes])
 
 def schema_digest(locus_names, alg=md5):
     hash = alg("\t".join(locus_names).encode())