Skip to content

Commit 1584639

Browse files
Will-Tylerjeromekelleher
authored andcommitted
Wait for preceding future before writing
1 parent 107174a commit 1584639

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

vcztools/vcf_writer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,7 @@ def write_vcf(
206206
drop_genotypes=drop_genotypes,
207207
no_update=no_update,
208208
executor=executor,
209+
preceding_future=preceding_future,
209210
)
210211
if preceding_future:
211212
concurrent.futures.wait((preceding_future,))
@@ -281,6 +282,7 @@ def write_vcf(
281282
drop_genotypes=drop_genotypes,
282283
no_update=no_update,
283284
executor=executor,
285+
preceding_future=preceding_future,
284286
)
285287
if preceding_future:
286288
concurrent.futures.wait((preceding_future,))
@@ -317,6 +319,7 @@ def c_chunk_to_vcf(
317319
drop_genotypes,
318320
no_update,
319321
executor: concurrent.futures.Executor,
322+
preceding_future: Optional[concurrent.futures.Future] = None,
320323
):
321324
chrom = None
322325
pos = None
@@ -431,6 +434,8 @@ def load_gt_phased():
431434

432435
futures.append(executor.submit(load_gt))
433436
futures.append(executor.submit(load_gt_phased))
437+
if preceding_future:
438+
futures.append(preceding_future)
434439
concurrent.futures.wait(futures)
435440

436441
ref = alleles[:, 0].astype("S")

0 commit comments

Comments
 (0)