fix: remove impossible try/catch

roedoejet · Mar 30, 2023 · 2db239a · 2db239a
1 parent 9cd84d8
commit 2db239a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/g2p/tests/test_transducer.py b/g2p/tests/test_transducer.py
@@ -244,6 +244,8 @@ def test_normalize_edges(self):
         bad_edges = [(0, None), (1, None), (2, 1)]
         self.assertEqual(normalize_edges(bad_edges), [(0, 1), (1, 1), (2, 1)])
         # Otherwise leave it as None
+        bad_edges = []
+        self.assertEqual(normalize_edges(bad_edges), bad_edges)
         bad_edges = [(0, None)]
         self.assertEqual(normalize_edges(bad_edges), bad_edges)
         bad_edges = [(0, None), (1, None)]

diff --git a/g2p/transducer/__init__.py b/g2p/transducer/__init__.py
@@ -71,10 +71,8 @@ def normalize_edges(
         if edge[1] is None:
             # if previous exists, use that, otherwise use following, otherwise None
             previous = [x for x in edges[:i] if x[1] is not None]
-            try:
-                following = [x for x in edges[i + 1 :] if x[1] is not None]
-            except IndexError:
-                following = None
+            # Note: if len(edges) == 1, edges[1:] is empty, not an IndexError
+            following = [x for x in edges[i + 1 :] if x[1] is not None]
             if previous:
                 edges[i] = (edge[0], previous[-1][1])
             elif following: