diff --git a/pybind11_stubgen/__init__.py b/pybind11_stubgen/__init__.py index f3a9215..89f7a9c 100644 --- a/pybind11_stubgen/__init__.py +++ b/pybind11_stubgen/__init__.py @@ -4,6 +4,7 @@ import logging import re from argparse import ArgumentParser, Namespace +from collections.abc import Sequence from pathlib import Path from pybind11_stubgen.parser.interface import IParser @@ -300,12 +301,12 @@ class Parser( return parser -def main(): +def main(argv: Sequence[str] | None = None) -> None: logging.basicConfig( level=logging.INFO, format="%(name)s - [%(levelname)7s] %(message)s", ) - args = arg_parser().parse_args(namespace=CLIArgs()) + args = arg_parser().parse_args(argv, namespace=CLIArgs()) parser = stub_parser_from_args(args) printer = Printer(invalid_expr_as_ellipses=not args.print_invalid_expressions_as_is)