Multiple dependent modules

[johnneave]

I am trying to build pyi files for multiple modules:
pybind11-stubgen a
pybind11-stubgen b
doesn't work because b has dependences on a

pybind11-stubgen used to have the option 'pybind11-stubgen a b' but that seems to have been removed. stubgen can process multiple modules but the output is inferior.

Any insight as to why the option was removed and if it can come back?

[victbr]

same issues. In my case, here the a is torch, and b is some torch cpp extension module

[victbr]

Find a workaround, in pybind11_stubgen/__init__.py, add import a in run, for example,

def run(
    parser: IParser,
    printer: Printer,
    module_name: str,
    out_dir: Path,
    sub_dir: Path | None,
    dry_run: bool,
    writer: Writer,
):
    import torch
    module = parser.handle_module(
        QualifiedName.from_str(module_name), importlib.import_module(module_name)
    )
    parser.finalize()

    if module is None:
        raise RuntimeError(f"Can't parse {module_name}")

    if dry_run:
        return

    out_dir.mkdir(exist_ok=True, parents=True)
    writer.write_module(module, printer, to=out_dir, sub_dir=sub_dir)