Lies to appease the type checker :/

lark-parser · Feb 25, 2024 · d86f1b3 · d86f1b3
1 parent e53a0ae
commit d86f1b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lark/visitors.py b/lark/visitors.py
@@ -156,11 +156,11 @@ def _transform_tree(self, tree):
         children = list(self._transform_children(tree.children))
         return self._call_userfunc(tree, children)
 
-    def transform(self, tree: Union[Tree[_Leaf_T], _Leaf_T]) -> Optional[_Return_T]:
+    def transform(self, tree: Tree[_Leaf_T]) -> _Return_T:
         "Transform the given tree, and return the final result"
         res = list(self._transform_children([tree]))
         if not res:
-            return
+            raise RuntimeError("Cannot discard the root node")
         assert len(res) == 1
         return res[0]