Make sure VcfHeaderNumber.value is called in VariantBuilder.

fulcrumgenomics · Jan 17, 2025 · 063b979 · 063b979
1 parent b0b4227
commit 063b979
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/fgpyo/vcf/builder.py b/fgpyo/vcf/builder.py
@@ -332,6 +332,9 @@ def add_info_header(
         """
         if field_type == VcfFieldType.FLAG:
             number = 0  # FLAGs always have number = 0
+        elif isinstance(number, VcfFieldNumber):
+            number = number.value
+
         header_line = f"##INFO=<ID={name},Number={number},Type={field_type.value}"
         if description is not None:
             header_line += f",Description={description}"
@@ -358,6 +361,9 @@ def add_format_header(
             number: the number of the field
             description: the description of the field
         """
+        if isinstance(number, VcfFieldNumber):
+            number = number.value
+
         header_line = f"##FORMAT=<ID={name},Number={number},Type={field_type.value}"
         if description is not None:
             header_line += f",Description={description}"