fix corner case

ablab · Apr 22, 2021 · 70f2dc0 · 70f2dc0
1 parent 3b517f6
commit 70f2dc0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/polya_verification.py b/src/polya_verification.py
@@ -64,10 +64,10 @@ def correct_read_info(self, read_exons, polya_info):
 
         if polyt_exon_count > 0 and polya_exon_count > 0:
             logger.debug("Both PolyA and PolyT fake terminal exons found: %d, %d" % (polya_exon_count, polyt_exon_count))
-            if polyt_exon_count + polya_exon_count == len(read_exons):
-                logger.debug("All exons seem to be consist of polyA/T")
-                polyt_exon_count -= 1
-                polya_exon_count -= 1
+        if polyt_exon_count + polya_exon_count == len(read_exons):
+            logger.debug("All exons seem to be consist of polyA/T")
+            polyt_exon_count -= 1
+            polya_exon_count -= 1
 
         if polya_exon_count > 0:
             polya_info.internal_polya_pos = shift_polya(read_exons, polya_exon_count, polya_info.internal_polya_pos)