get rid of side-effect

ablab · Nov 15, 2024 · 7d495f7 · 7d495f7
1 parent 099bdc8
commit 7d495f7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/intron_graph.py b/src/intron_graph.py
@@ -209,6 +209,8 @@ def signleton_dead_start(self, v):
 
     def get_outgoing(self, intron, v_type=None):
         res = []
+        if intron not in self.outgoing_edges:
+            return res
         if v_type is None:
             for v in self.outgoing_edges[intron]:
                 if v[0] >= 0:
@@ -221,6 +223,8 @@ def get_outgoing(self, intron, v_type=None):
 
     def get_incoming(self, intron, v_type=None):
         res = []
+        if intron not in self.incoming_edges:
+            return res
         if v_type is None:
             for v in self.incoming_edges[intron]:
                 if v[0] >= 0: