diff --git a/neat/read_simulator/utils/options.py b/neat/read_simulator/utils/options.py index a45f8664..fd05b6de 100644 --- a/neat/read_simulator/utils/options.py +++ b/neat/read_simulator/utils/options.py @@ -439,7 +439,7 @@ def log_configuration(self): if self.parallel_mode == 'size': _LOG.info(f'Splitting reference into chunks.') - _LOG.info(f' - splitting input into size {self.size}') + _LOG.info(f' - splitting input into size {self.parallel_block_size}') elif self.parallel_mode == 'contig': _LOG.info(f'Splitting input by contig.') if not self.cleanup_splits or self.reuse_splits: diff --git a/neat/read_simulator/utils/split_inputs.py b/neat/read_simulator/utils/split_inputs.py index 82f63cef..ad3a42f9 100644 --- a/neat/read_simulator/utils/split_inputs.py +++ b/neat/read_simulator/utils/split_inputs.py @@ -67,7 +67,7 @@ def main(options: Options, reference_index: dict) -> tuple[dict, int]: # We'll keep track of chunks by contig, to help us out later split_fasta_dict: dict[str, dict[tuple[int, int], Path]] = {key: {} for key in reference_index.keys()} for contig, seq_record in reference_index.items(): - if options.mode == "contig": + if options.parallel_mode == "contig": stem = f"{idx:0{pad}d}__{contig}" fa = options.splits_dir / f"{stem}.fa.gz" write_fasta(contig, seq_record.seq.upper(), fa) @@ -75,7 +75,7 @@ def main(options: Options, reference_index: dict) -> tuple[dict, int]: idx += 1 written += 1 else: - for start, subseq in chunk_record(seq_record.seq.upper(), options.size, overlap): + for start, subseq in chunk_record(seq_record.seq.upper(), options.parallel_block_size, overlap): stem = f"{idx:0{pad}d}__{contig}" fa = options.splits_dir / f"{stem}.fa.gz" write_fasta(contig, subseq, fa)