Fix tail a bug when a schema has extra columns

EnzoAndree · Feb 4, 2021 · 8dc6f45 · 8dc6f45
1 parent 7ba0c40
commit 8dc6f45
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/bin/fastmlst b/bin/fastmlst
@@ -48,7 +48,7 @@ def runMLST(margument):
 
 
 if __name__ == '__main__':
-    V = '%(prog)s v0.0.10'
+    V = '%(prog)s v0.0.11'
     parser = argparse.ArgumentParser()
     parser.add_argument(type=str, nargs='*', dest='genomes')
     parser.add_argument('-t', '--threads', type=int, default=cpu_count(),

diff --git a/fastmlst/mlst.py b/fastmlst/mlst.py
@@ -220,12 +220,7 @@ def str_allelic_profile(self, ):
             dictofrows = {'Genome': self.beautiname,
                           'Scheme': self.scheme,
                           'ST': self.ST.index.values[0]}
-            for i in sorted(self.score['scheme'].keys()):
-                out = '{0}({1}){2}'.format(i, self.score['scheme'][i],
-                                           self.sep)
-                dictofrows[i] = self.score['scheme'][i]
-                output += out
-            for i in self.ST.iloc[: , [self.number_alleles, ]]:
+            for i in self.ST:
                 out = '{0}({1}){2}'.format(i, self.ST[i].values[0],
                                            self.sep)
                 dictofrows[i] = self.ST[i].values[0]

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='fastmlst',
-    version='0.0.10',
+    version='0.0.11',
     description='A multi-core tool for multilocus sequence typing of draft genome assemblies using PubMLST typing schemes',
     url='https://github.com/EnzoAndree/FastMLST',
     author='Enzo Guerrero-Araya',