make sure has_vaccine_peptide is always present

openvax · Aug 18, 2020 · 66fa687 · 66fa687
1 parent 7258edb
commit 66fa687
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/vaxrank/vaxrank_results.py b/vaxrank/vaxrank_results.py
@@ -111,7 +111,6 @@ def variant_properties(self, gene_pathway_check=None):
                     isovar_result.predicted_effect_modifies_protein_sequence),
                 ('rna_support',
                     isovar_result.has_mutant_protein_sequence_from_rna),
-                ('mhc_binder', False),
                 ('gene_name', gene_name),
             ))
             if gene_pathway_check is not None:
@@ -120,8 +119,8 @@ def variant_properties(self, gene_pathway_check=None):
 
             # TODO:
             #  compute MHC binder status for variants that don't have RNA support
-            if variant in self.variant_to_vaccine_peptides_dict:
-                variant_dict['mhc_binder'] = \
-                    variant_dict["has_vaccine_peptide"] = True
+            variant_dict['mhc_binder'] = \
+                variant_dict["has_vaccine_peptide"] = \
+                    variant in self.variant_to_vaccine_peptides_dict
             variant_properties_list.append(variant_dict)
         return variant_properties_list