diff options
Diffstat (limited to 'src')
| -rwxr-xr-x | src/python/quikr_train | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/python/quikr_train b/src/python/quikr_train index 3096d14..5a8ec69 100755 --- a/src/python/quikr_train +++ b/src/python/quikr_train @@ -32,14 +32,14 @@ def main(): if not os.path.isfile(args.input): parser.error( "Input database not found") - # call the quikr train function, save the output with np.save - matrix = quikr.train_matrix(args.input, args.kmer) - if args.kmer is None: kmer = 6 else: kmer = args.kmer + # call the quikr train function, save the output with np.save + matrix = quikr.train_matrix(args.input, kmer) + if args.compress: output_file = gzip.open(args.output, "wb") else: |