Fix the Tools/peg_generator/scripts/benchmark.py script (GH-19848)

diff --git a/Tools/peg_generator/pegen/build.py b/Tools/peg_generator/pegen/build.py
index 94248ffd9431c4d63adfe95f145834f8ed2d4d7b..d33dd049d63c7e94c588b3aa2b6a38bf32b55831 100644 (file)
--- a/Tools/peg_generator/pegen/build.py
+++ b/Tools/peg_generator/pegen/build.py
@@ -15,7 +15,7 @@ from pegen.parser_generator import ParserGenerator
 from pegen.python_generator import PythonParserGenerator
 from pegen.tokenizer import Tokenizer
 
-MOD_DIR = pathlib.Path(__file__).parent
+MOD_DIR = pathlib.Path(__file__).resolve().parent
 
 
 def get_extra_flags(compiler_flags: str, compiler_py_flags_nodist: str) -> List[str]:

diff --git a/Tools/peg_generator/scripts/benchmark.py b/Tools/peg_generator/scripts/benchmark.py
index bc751156e897221e5b13826f565fe24806684d09..6b4287cd8cecc67f8d4fcd67c68325be0c7a2234 100644 (file)
--- a/Tools/peg_generator/scripts/benchmark.py
+++ b/Tools/peg_generator/scripts/benchmark.py
@@ -11,7 +11,7 @@ import memory_profiler
 
 sys.path.insert(0, os.getcwd())
 from peg_extension import parse
-from pegen.build import build_parser_and_generator
+from pegen.build import build_c_parser_and_generator
 from scripts.test_parse_directory import parse_directory
 
 argparser = argparse.ArgumentParser(
@@ -93,8 +93,9 @@ def run_benchmark_stdlib(subcommand, parser):
     modes = {"compile": 2, "parse": 1, "check": 0}
     extension = None
     if parser == "pegen":
-        extension = build_parser_and_generator(
+        extension = build_c_parser_and_generator(
             "../../Grammar/python.gram",
+            "../../Grammar/Tokens",
             "peg_extension/parse.c",
             compile_extension=True,
             skip_actions=False,