# HG changeset patch
# User bornea
# Date 1472343909 14400
# Node ID 26b5bd96332e9935521d5f17db88ec299cd8dc8f
# Parent  8031a47f67c6cf813d411c723314beb00360cce5
Uploaded

diff -r 8031a47f67c6 -r 26b5bd96332e SAINT_preprocessing.py
--- a/SAINT_preprocessing.py	Wed Jun 08 10:36:32 2016 -0400
+++ b/SAINT_preprocessing.py	Sat Aug 27 20:25:09 2016 -0400
@@ -110,6 +110,7 @@
     db_len = len(data_lines)
     seqlength = 0
     count = 0
+    last_line = data_lines[-1]
     for data_line in data_lines:
         if ">sp" in data_line:
             namer = data_line.split("|")[2]
@@ -124,6 +125,8 @@
                 while ">sp" not in data_lines[match]:
                     if match <= db_len:
                         seqlength = seqlength + len(data_lines[match].strip())
+                        if data_lines[match] == last_line:
+                            break
                         match = match + 1
                     else:
                         break
@@ -140,6 +143,8 @@
             while ">sp" not in data_lines[match]:
                 if match <= db_len:
                     seqlength = seqlength + len(data_lines[match].strip())
+                    if data_lines[match] == last_line:
+                        break
                     match = match + 1
                 else:
                     break