From 852606fc2fcbe387b9968885c71956593d6690d4 Mon Sep 17 00:00:00 2001
From: mark0428 <Chenkai@ChenkaideMacBook-Pro.local>
Date: Mon, 29 Nov 2021 22:06:19 -0800
Subject: [PATCH] fix typo

---
 src/AMPlify.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/AMPlify.py b/src/AMPlify.py
index 2a2f2ba..c04dd5a 100755
--- a/src/AMPlify.py
+++ b/src/AMPlify.py
@@ -191,7 +191,7 @@ def main():
         seq_id.append(str(seq_record.id))
         peptide.append(str(seq_record.seq))
     
-    # look for indices for valid sequences
+    # look for indices of valid sequences
     valid_ix = []
     for i in range(len(peptide)):
         if len(peptide[i]) <= 200 and len(peptide[i]) >= 2 and set(peptide[i])-set(aa) == set():
@@ -264,7 +264,7 @@ def main():
                                         'Sequence': peptide,
                                         'Score': y_score,
                                         'Prediction': y_class,
-                                        'Attention': [a for a in attention]})
+                                        'Attention': attention})
                 else:
                     out = pd.DataFrame({'Sequence_ID':seq_id,
                                         'Sequence': peptide,