Use allele.organism instead of str for allele conversion

FRED-2 · Apr 22, 2020 · aa71ae3 · aa71ae3
1 parent bc97853
commit aa71ae3
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/Fred2/EpitopePrediction/ANN.py b/Fred2/EpitopePrediction/ANN.py
@@ -149,9 +149,9 @@ def _represent(self, allele):
             :return: str
             """
             if isinstance(allele, MouseAllele):
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype.upper(), allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype.upper(), allele.subtype)
             else:
-                return "HLA-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts FRED2s internal allele representation into the format required by mhcnuggets
         def convert_alleles(self, alleles):
@@ -308,9 +308,9 @@ def _represent(self, allele):
             :return: str
             """
             if isinstance(allele, MouseAllele):
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
             else:
-                return "HLA-%s%s:%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s:%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts the internal mhcnuggets-class-1 HLA representation back into a FRED2 representation
         def revert_allele_repr(self, name):
@@ -408,12 +408,12 @@ def _represent(self, allele):
             """
             if isinstance(allele, MouseAllele):
                 # expects H-2-XXx
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype.upper(), allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype.upper(), allele.subtype)
             elif isinstance(allele, CombinedAllele):
-                return "HLA-%s%s%s-%s%s%s" % (allele.alpha_locus, allele.alpha_supertype, allele.alpha_subtype,
+                return "%s-%s%s%s-%s%s%s" % (allele.organism, allele.alpha_locus, allele.alpha_supertype, allele.alpha_subtype,
                                         allele.beta_locus, allele.beta_supertype, allele.beta_subtype)
             else:
-                return "HLA-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts FRED2s internal allele representation into the format required by mhcnuggets
         def convert_alleles(self, alleles):
@@ -599,12 +599,12 @@ def _represent(self, allele):
             """
             if isinstance(allele, MouseAllele):
                 # expects H-2-XXx
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype.upper(), allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype.upper(), allele.subtype)
             elif isinstance(allele, CombinedAllele):
-                return "HLA-%s%s:%s-%s%s:%s" % (allele.alpha_locus, allele.alpha_supertype, allele.alpha_subtype,
+                return "%s-%s%s:%s-%s%s:%s" % (allele.organism, allele.alpha_locus, allele.alpha_supertype, allele.alpha_subtype,
                                         allele.beta_locus, allele.beta_supertype, allele.beta_subtype)
             else:
-                return "HLA-%s%s:%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s:%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts the internal mhcnuggets-class-2 representation back into a FRED2 representation
         def revert_allele_repr(self, name):
@@ -684,9 +684,9 @@ def _represent(self, allele):
             :return: str
             """
             if isinstance(allele, MouseAllele):
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
             else:
-                return "HLA-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts FRED2s internal allele representation into the format required by MHCFlurry
         def convert_alleles(self, alleles):
@@ -825,9 +825,9 @@ def _represent(self, allele):
             :return: str
             """
             if isinstance(allele, MouseAllele):
-                return "H-2-%s%s%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s%s%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
             else:
-                return "HLA-%s*%s:%s" % (allele.locus, allele.supertype, allele.subtype)
+                return "%s-%s*%s:%s" % (allele.organism, allele.locus, allele.supertype, allele.subtype)
 
         # Converts the internal MHCFlurry representation back into a FRED2 representation
         def revert_allele_repr(self, name):