Skip to content

Commit

Permalink
load read_ids when --resume is passed
Browse files Browse the repository at this point in the history
  • Loading branch information
andrewprzh committed Jul 11, 2024
1 parent d8115ad commit 5fd20ab
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/dataset_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,11 @@ def collect_reads_in_parallel(sample, chr_id, args):
for g in open(group_file):
read_grouper.read_groups.add(g.strip())
alignment_stat_counter = EnumStats(bamstat_file)
loader = BasicReadAssignmentLoader(save_file)
while loader.has_next():
for read_assignment in loader.get_next():
if read_assignment is None: continue
processed_reads.append(read_assignment.read_id)
logger.info("Loaded data for " + chr_id)
return read_grouper.read_groups, alignment_stat_counter, processed_reads
else:
Expand Down

0 comments on commit 5fd20ab

Please sign in to comment.