Fix type

opensanctions · Nov 18, 2024 · 17b4c3d · 17b4c3d
1 parent 073d1b2
commit 17b4c3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nomenklatura/statement/serialize.py b/nomenklatura/statement/serialize.py
@@ -163,7 +163,7 @@ def __init__(self, fh: TextIO) -> None:
         self.fh = fh
         self.writer = csv.writer(self.fh, dialect=csv.unix_dialect)
         self.writer.writerow(CSV_COLUMNS)
-        self._batch: List[List[str]] = []
+        self._batch: List[List[Optional[str]]] = []
 
     def write(self, stmt: S) -> None:
         row = stmt.to_csv_row()
@@ -186,7 +186,7 @@ def __init__(self, fh: TextIO) -> None:
             dialect=csv.unix_dialect,
             quoting=csv.QUOTE_MINIMAL,
         )
-        self._batch: List[List[str]] = []
+        self._batch: List[List[Optional[str]]] = []
 
     def write(self, stmt: S) -> None:
         row = stmt.to_csv_row()