Fixes according to review notes (make tree_class optional)

lark-parser · Sep 25, 2023 · b8b69c7 · b8b69c7
1 parent b7f1ebb
commit b8b69c7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/lark/parsers/earley.py b/lark/parsers/earley.py
@@ -9,7 +9,7 @@
 is explained here: https://lark-parser.readthedocs.io/en/latest/_static/sppf/sppf.html
 """
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING, Callable, Optional, Sequence, Any
 from collections import deque
 
 from ..lexer import Token
@@ -30,7 +30,8 @@ class Parser:
     debug: bool
 
     def __init__(self, lexer_conf: 'LexerConf', parser_conf: 'ParserConf', term_matcher: Callable,
-                 resolve_ambiguity: bool=True, debug: bool=False, tree_class: type=Tree, ordered_sets: bool=True):
+                 resolve_ambiguity: bool=True, debug: bool=False,
+                 tree_class: Optional[type]=Tree, ordered_sets: bool=True):
         analysis = GrammarAnalyzer(parser_conf)
         self.lexer_conf = lexer_conf
         self.parser_conf = parser_conf

diff --git a/lark/parsers/xearley.py b/lark/parsers/xearley.py
@@ -14,7 +14,7 @@
 Earley's power in parsing any CFG.
 """
 
-from typing import TYPE_CHECKING, Callable
+from typing import TYPE_CHECKING, Callable, Optional, Sequence, Any
 from collections import defaultdict
 
 from ..tree import Tree
@@ -30,7 +30,7 @@
 class Parser(BaseParser):
     def __init__(self, lexer_conf: 'LexerConf', parser_conf: 'ParserConf', term_matcher: Callable,
                  resolve_ambiguity: bool=True, complete_lex: bool=False, debug: bool=False,
-                 tree_class: type=Tree, ordered_sets: bool=True):
+                 tree_class: Optional[type]=Tree, ordered_sets: bool=True):
         BaseParser.__init__(self, lexer_conf, parser_conf, term_matcher, resolve_ambiguity,
                             debug, tree_class, ordered_sets)
         self.ignore = [Terminal(t) for t in lexer_conf.ignore]