Add TODO and notes

szhan · Feb 26, 2024 · 4d8ca10 · 4d8ca10
1 parent 03ffab6
commit 4d8ca10
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -756,9 +756,13 @@ def compute_allele_frequency(gt_probs, allele=1):
 
 def write_vcf(ref_ts, impdata, out_file, chr_name="1"):
     """
+    Print imputation results in VCF format, following the VCF output of BEAGLE 4.1.
+
     x = number of imputed sites
     n = number of diploid individuals
 
+    TODO: Print VCF records for genotyped sites.
+
     :param tskit.TreeSequence ref_ts: Tree sequence with reference haplotypes.
     :param ImpData impdata: Object containing imputation data.
     :param str out_file: Path to output VCF file.