diff --git a/tools/precompiler/precompiler.py b/tools/precompiler/precompiler.py index ae7339abfe..310f2eb097 100644 --- a/tools/precompiler/precompiler.py +++ b/tools/precompiler/precompiler.py @@ -24,8 +24,8 @@ def _create_parser() -> "argparse.Namespace": parser = argparse.ArgumentParser(fromfile_prefix_chars="@") - parser.add_argument("--invalidation_mode") - parser.add_argument("--optimize", type=int) + parser.add_argument("--invalidation_mode", default="CHECKED_HASH") + parser.add_argument("--optimize", type=int, default=-1) parser.add_argument("--python_version") parser.add_argument("--src", action="append", dest="srcs") @@ -40,10 +40,10 @@ def _create_parser() -> "argparse.Namespace": def _compile(options: "argparse.Namespace") -> None: try: - invalidation_mode = getattr( - py_compile.PycInvalidationMode, options.invalidation_mode.upper() - ) - except AttributeError as e: + invalidation_mode = py_compile.PycInvalidationMode[ + options.invalidation_mode.upper() + ] + except KeyError as e: raise ValueError( f"Unknown PycInvalidationMode: {options.invalidation_mode}" ) from e