From c5e0e05b5b29f8e94ddbb02d92713f9d4d19350c Mon Sep 17 00:00:00 2001
From: Genomics team in Google Brain <no-reply@google.com>
Date: Mon, 15 Mar 2021 19:29:52 -0700
Subject: [PATCH] Add per sample stats for training examples.

PiperOrigin-RevId: 363087311
---
 nucleus/util/variantcall_utils.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/nucleus/util/variantcall_utils.py b/nucleus/util/variantcall_utils.py
index c191f1d..c85f7ee 100644
--- a/nucleus/util/variantcall_utils.py
+++ b/nucleus/util/variantcall_utils.py
@@ -190,6 +190,12 @@ def get_min_dp(variant_call):
       variant_call.info, 'MIN_DP', is_single_field=True)
 
 
+def set_bam_fname(variant_call, bam_fname):
+  """Sets 'BAM_FNAME' field of the VariantCall."""
+  return struct_utils.set_string_field(variant_call.info, 'BAM_FNAME',
+                                       bam_fname)
+
+
 def has_genotypes(variant_call):
   """Returns True iff the VariantCall has one or more called genotypes.