perf: parallelize checksum generation for contig

bento-platform · Nov 16, 2023 · bd02ffc · bd02ffc
1 parent e284667
commit bd02ffc
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/fasta_checksum_utils/_contig.py b/fasta_checksum_utils/_contig.py
@@ -1,3 +1,4 @@
+import asyncio
 import pysam
 from typing import Generator
 from .algorithms import ChecksumAlgorithm
@@ -23,8 +24,4 @@ def gen_sequence() -> Generator[bytes, None, None]:
                 .encode("ascii")
             )
 
-    r = []
-    for a in algorithms:
-        r.append(await a.checksum_sequence(gen_sequence()))
-
-    return tuple(r)
+    return tuple(await asyncio.gather(*(a.checksum_sequence(gen_sequence()) for a in algorithms)))