option: keep only MANE select transcripts

ProGenNo · May 15, 2024 · 7dc1791 · 7dc1791
1 parent 7eec65c
commit 7dc1791
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/get_reference_ENST.py b/src/get_reference_ENST.py
@@ -36,7 +36,8 @@
     chr = prot['description'].split(':',3)[2]
 
     if chr in CHROMOSOMES:
-        if ('MANE_Select' in annotations_db[trID].attributes['tag']):
+        transcript_feature = annotations_db[trID]
+        if (('tag' in transcript_feature.attributes) and ('MANE_Select' in transcript_feature.attributes['tag'])):
                 result_data.append([chr,trID])
 
 result_df = pd.DataFrame(data=result_data, columns=['chromosome', 'transcriptID'])