Small code style improvements for contig stitcher

cfe-lab · Nov 30, 2023 · 7aec90f · 7aec90f
1 parent f1d88db
commit 7aec90f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/micall/core/contig_stitcher.py b/micall/core/contig_stitcher.py
@@ -1,4 +1,4 @@
-from typing import Iterable, Optional, Tuple, List
+from typing import Iterable, Optional, Tuple, List, Dict
 from collections import deque, defaultdict
 from dataclasses import dataclass
 from math import ceil
@@ -530,7 +530,9 @@ def stitch_contigs(contigs: Iterable[GenotypedContig]) -> Iterable[AlignedContig
 
 def stitch_consensus(contigs: Iterable[GenotypedContig]) -> Iterable[GenotypedContig]:
     contigs = list(stitch_contigs(contigs))
-    consensus_parts = defaultdict(list) # group_ref -> List[AlignedContig]
+
+    GroupRef = str
+    consensus_parts: Dict[GroupRef, AlignedContig] = defaultdict(list)
 
     for contig in contigs:
         if isinstance(contig, AlignedContig):