fix a bug when input < 50

BigDataBiology · Jun 5, 2020 · f96d173 · f96d173
1 parent f6212f4
commit f96d173
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/gmgc_finder/main.py b/gmgc_finder/main.py
@@ -87,6 +87,15 @@ def split(handle):
                     SeqIO.write(split_fasta, output_dir + '/split_{}.fna'.format(index), 'fasta')
                 else:
                     SeqIO.write(split_fasta, output_dir + '/split_{}.faa'.format(index), 'fasta')
+        else:
+            index += 1
+            for seq_record in records:
+                rec1 = SeqRecord(Seq(str(seq_record.seq)), id=seq_record.id, description='')
+                split_fasta.append(rec1)
+            if is_dna is True:
+                SeqIO.write(split_fasta, output_dir + '/split_{}.fna'.format(index), 'fasta')
+            else:
+                SeqIO.write(split_fasta, output_dir + '/split_{}.faa'.format(index), 'fasta')
         return index
 
     if not os.path.exists(gene_path):