fixing partition genes bug

VoSeq · Mar 19, 2015 · 3493d04 · 3493d04
1 parent 0747680
commit 3493d04
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/voseq/create_dataset/utils.py b/voseq/create_dataset/utils.py
@@ -62,8 +62,6 @@ def get_charset_block(self):
         bp_count_start = 0
         bp_count_end = 0
         self.gene_codes.sort()
-        print(">>>>>>>sel.gene_codes", self.gene_codes)
-        print(">>>>>>>sel.gene_codes_and_lengths", self.gene_codes_and_lengths)
         for gene in self.gene_codes_and_lengths:
             bp_count_end += self.gene_codes_and_lengths[gene]
             line = '    charset ' + gene + ' = ' + str(
@@ -73,8 +71,8 @@ def get_charset_block(self):
         return charset_block
 
     def get_partitions_block(self):
-        line = 'partition GENES = ' + str(len(self.gene_codes))
-        line += ': ' + ', '.join(self.gene_codes) + ';\n'
+        line = 'partition GENES = ' + str(len(self.gene_codes_and_lengths))
+        line += ': ' + ', '.join([i for i in self.gene_codes_and_lengths]) + ';\n'
         line += '\nset partition = GENES;\n'
         return [line]