fix (splitFasta): variant peptide not split correclty if derived from…

… fusion within the same gene
uclahs-cds · Feb 19, 2025 · fed5113 · fed5113
1 parent bc73cfd
commit fed5113
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/moPepGen/aa/VariantPeptideLabel.py b/moPepGen/aa/VariantPeptideLabel.py
@@ -189,9 +189,10 @@ def from_variant_peptide(peptide:AminoAcidSeqRecord,
                 var_ids = {
                     first_gene_id: variant_id.first_variants
                         + [variant_id.fusion_id] \
-                        + variant_id.peptide_variants,
-                    second_gene_id: variant_id.second_variants
+                        + variant_id.peptide_variants
                 }
+                if second_gene_id != first_gene_id:
+                    var_ids[second_gene_id] = variant_id.second_variants
                 tx_id = first_tx_id
 
             elif isinstance(variant_id, pi.BaseVariantPeptideIdentifier):