From de379a797a5c50f58cf489dddc0bf558aadb8a30 Mon Sep 17 00:00:00 2001
From: tfenne <tfenne@tfenne.com>
Date: Fri, 17 Jan 2025 14:10:17 -0700
Subject: [PATCH] Make mypy happy.

---
 fgpyo/vcf/builder.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/fgpyo/vcf/builder.py b/fgpyo/vcf/builder.py
index a4cda71..2a9a57e 100644
--- a/fgpyo/vcf/builder.py
+++ b/fgpyo/vcf/builder.py
@@ -331,11 +331,13 @@ def add_info_header(
             version: the version of the field
         """
         if field_type == VcfFieldType.FLAG:
-            number = 0  # FLAGs always have number = 0
+            num = "0"  # FLAGs always have number = 0
         elif isinstance(number, VcfFieldNumber):
-            number = number.value
+            num = number.value
+        else:
+            num = str(number)
 
-        header_line = f"##INFO=<ID={name},Number={number},Type={field_type.value}"
+        header_line = f"##INFO=<ID={name},Number={num},Type={field_type.value}"
         if description is not None:
             header_line += f",Description={description}"
         if source is not None:
@@ -362,9 +364,11 @@ def add_format_header(
             description: the description of the field
         """
         if isinstance(number, VcfFieldNumber):
-            number = number.value
+            num = number.value
+        else:
+            num = str(number)
 
-        header_line = f"##FORMAT=<ID={name},Number={number},Type={field_type.value}"
+        header_line = f"##FORMAT=<ID={name},Number={num},Type={field_type.value}"
         if description is not None:
             header_line += f",Description={description}"
         header_line += ">"