Faster count_variant_alleles in Tajimas_D

sgkit-dev · Sep 9, 2023 · 69f9293 · 69f9293
1 parent 9e8bd1e
commit 69f9293
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sgkit/stats/popgen.py b/sgkit/stats/popgen.py
@@ -430,7 +430,11 @@ def Tajimas_D(
            [1.10393559, 1.10393559]])
     """
     ds = define_variable_if_absent(
-        ds, variables.variant_allele_count, variant_allele_count, count_variant_alleles
+        ds,
+        variables.variant_allele_count,
+        variant_allele_count,
+        count_variant_alleles,
+        using=variables.call_genotype,
     )
     ds = define_variable_if_absent(
         ds, variables.stat_diversity, stat_diversity, diversity