Update main.py

fixed problem with top parameter
ahtosalumets · Jan 22, 2019 · fb3442c · fb3442c
1 parent 8e00af8
commit fb3442c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -64,7 +64,7 @@ def read_file(file, top):
     with open(file, 'r') as f:
         content = f.readlines()
     n_seqs = len(content) // 2
-    if top < n_seqs:
+    if top != 'all' and top < n_seqs:
         n_seqs = top
     for i in range(0, 2*n_seqs-1, 2):
         seq_name = content[i].strip().replace('>', '')