Update mlst.py

a minnor bug fixed
EnzoAndree · Mar 8, 2021 · 1942951 · 1942951
1 parent fd57aac
commit 1942951
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/fastmlst/mlst.py b/fastmlst/mlst.py
@@ -337,15 +337,17 @@ def mlstex(self, ):
             header = self.beautiname + ' '
             concatenatedseq = ''
             for genename in self.name_alleles:
-                header += genename + '_'
-                concatenatedseq += self.alleles[genename].seq
+                if genename in self.alleles.keys():
+                    header += genename + '_'
+                    concatenatedseq += self.alleles[genename].seq
             record_out = SeqRecord(concatenatedseq, id=header.strip('_'),
                                    description='Concatenated Sequences of MLST ' +
                                    'from ' + self.beautiname)
         else:
             concatenatedseq = ''
             for genename in self.name_alleles:
-                concatenatedseq += self.alleles[genename].seq
+                if genename in self.alleles.keys():
+                    concatenatedseq += self.alleles[genename].seq
             record_out = SeqRecord(concatenatedseq, id=self.beautiname,
                 description='')
         return record_out