fixed kmer argument bug

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: