From 9889011181c76f5ea562b3468588ce34f010ab62 Mon Sep 17 00:00:00 2001 From: Shing Zhan Date: Sun, 25 Feb 2024 20:38:10 +0000 Subject: [PATCH] Add individual names to column line --- python/tests/beagle_numba.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py index 16c60c8a90..5b723d1b5c 100644 --- a/python/tests/beagle_numba.py +++ b/python/tests/beagle_numba.py @@ -746,7 +746,6 @@ def write_vcf(ref_ts, impdata, out_file, chr_name="1"): + 'Description="Estimated Genotype Probability">', ] _COLUMN_NAMES = [ - "#", "CHROM", "POS", "ID", @@ -760,8 +759,10 @@ def write_vcf(ref_ts, impdata, out_file, chr_name="1"): with open(out_file, "w") as f: for line in _HEADER: f.write(line + "\n") - for col_name in _COLUMN_NAMES: - f.write(col_name + "\t") + col_str = "#" + col_str += "\t".join(_COLUMN_NAMES) + col_str += "\t".join(impdata.individual_names) + f.write(col_str + "\n") for i in range(impdata.num_sites): a1, ap1, a2, ap2 = impdata.get_alleles_at_site(i) gt_probs, dosages = compute_individual_scores(a1, ap1, a2, ap2)