Skip to content

Commit

Permalink
debugged filter_groups.py
Browse files Browse the repository at this point in the history
minimum_nr_species parameter was hard coded as 3 and not using the argument specified
  • Loading branch information
natashaglover authored Apr 16, 2020
1 parent 8d7bfc8 commit 5717807
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions filter_groups.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,9 +47,9 @@ def get_matching_files(input_dir, pattern):
yield f


def convert_files(input_dir, output_dir, pattern='.fa', nr_processes=None):
def convert_files(input_dir, output_dir, min_nr_species, pattern='.fa', nr_processes=None):
os.makedirs(output_dir, exist_ok=True)
args_iter = ((x.path, os.path.join(output_dir, x.name), 3) for x in get_matching_files(input_dir, pattern))
args_iter = ((x.path, os.path.join(output_dir, x.name), min_nr_species) for x in get_matching_files(input_dir, pattern))
with multiprocessing.Pool(processes=nr_processes) as mp:
logger.info("converting files in {} with {} processes in parallel"
.format(input_dir, nr_processes if nr_processes else os.cpu_count()))
Expand Down Expand Up @@ -101,4 +101,4 @@ class SeveralSequencePerSpeciesException(Exception):
logging.basicConfig(level=level, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
logger.debug("Concatenate alignments: arguments: {}".format(conf))

convert_files(conf.input, conf.output, pattern=conf.ext, nr_processes=conf.threads)
convert_files(conf.input, conf.output, conf.min_nr_species, pattern=conf.ext, nr_processes=conf.threads)

0 comments on commit 5717807

Please sign in to comment.