FIX : Using proper class for proper gene type when reading the pangen…

…ome hdf5 file
labgem · Nov 28, 2019 · c9b2e85 · c9b2e85
1 parent 322e7db
commit c9b2e85
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-1.0.10
+1.0.11
diff --git a/ppanggolin/formats/readBinaries.py b/ppanggolin/formats/readBinaries.py
@@ -10,7 +10,7 @@
 import tables
 
 #local libraries
-from ppanggolin.genome import Organism, Gene
+from ppanggolin.genome import Organism, Gene, RNA
 
 
 def getNumberOfOrganisms(pangenome):
@@ -92,7 +92,11 @@ def readOrganism(pangenome, orgName, contigDict, link = False):
             if link:#if the gene families are already computed/loaded the gene exists.
                 gene = pangenome.getGene(row["ID"].decode())
             else:#else creating the gene.
-                gene = Gene(row["ID"].decode())
+                gene_type = row["type"].decode()
+                if gene_type == "CDS":
+                    gene = Gene(row["ID"].decode())
+                elif "RNA" in gene_type:
+                    gene = RNA(row["ID"].decode())
             try:
                 local = row["local"].decode()
             except ValueError: