Merge branch 'master' of github.com:malariagen/malariagen-data-python

malariagen · Jul 29, 2021 · f3c3404 · f3c3404
2 parents e5a8a8a + b27624b
commit f3c3404
Show file tree

Hide file tree

Showing 2 changed files with 185 additions and 149 deletions.
diff --git a/malariagen_data/ag3.py b/malariagen_data/ag3.py
@@ -796,6 +796,9 @@ def snp_allele_frequencies(
         for coh, query in cohorts.items():
             # locate samples
             loc_coh = df_meta.eval(query).values
+            n_samples = np.count_nonzero(loc_coh)
+            if n_samples == 0:
+                raise ValueError(f"no samples for cohort {coh!r}")
             gt_coh = np.compress(loc_coh, gt, axis=1)
             # count alleles
             ac_coh = allel.GenotypeArray(gt_coh).count_alleles(max_allele=3)