From ee089c90b04c6a00db168727b0972c9484b3f8b5 Mon Sep 17 00:00:00 2001 From: Calvin Morrison Date: Mon, 31 Mar 2014 14:08:16 -0400 Subject: proper sizes with custom mers --- src/score_mers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/score_mers.py b/src/score_mers.py index fc1e81c..ef90be1 100755 --- a/src/score_mers.py +++ b/src/score_mers.py @@ -249,11 +249,11 @@ def score_all_combinations(mers): fh = open(output_file, 'wb') write_header(fh) - max_size = max_select+1 - if len(mers) < max_select + 1: + max_size = max_select + if len(mers) < max_select: max_size = len(mers) + 1 - for select_n in range(1, max_size ): + for select_n in range(1, max_size + 1 ): print "scoring size ", select_n, t = time.time() scores_it = p.imap_unordered(score, combinations(mers, select_n), chunksize=8192) -- cgit v1.2.3