Add per sample stats for training examples.

PiperOrigin-RevId: 363087311
google · Mar 16, 2021 · c5e0e05 · c5e0e05
1 parent a041ea8
commit c5e0e05
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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.