Add gt probs

szhan · Feb 25, 2024 · f022b33 · f022b33
1 parent 73abd52
commit f022b33
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tests/beagle_numba.py b/python/tests/beagle_numba.py
@@ -787,7 +787,9 @@ def write_vcf(ref_ts, impdata, out_file, chr_name="1"):
             format_str = ""
             for j in range(impdata.num_individuals):
                 format_str += impdata.alleles[j] + "|" + impdata.alleles[j + 1] + ":"
-                # format_str += round(gt_probs[j], 2) + ":"
-                format_str += round(dosages[j], 2)
+                format_str += str(round(dosages[j], 2)) + ":"
+                format_str += (
+                    ",".join([str(round(gt_probs[j, k], 2)) for k in range(3)]) + ":"
+                )
                 format_str += "\t"
             line_str += format_str + "\n"