From 9be3f0c388e661a3dec2cc1afeec9b109bfd361a Mon Sep 17 00:00:00 2001 From: dillydill123 Date: Fri, 17 Apr 2026 10:32:13 -0400 Subject: [PATCH] Add direct_pyi_files and transitive_pyi_files to py_pyo3_library PyInfo provider --- extensions/pyo3/private/pyo3.bzl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/extensions/pyo3/private/pyo3.bzl b/extensions/pyo3/private/pyo3.bzl index 865ae7a83d..577e677d9d 100644 --- a/extensions/pyo3/private/pyo3.bzl +++ b/extensions/pyo3/private/pyo3.bzl @@ -131,6 +131,7 @@ def _py_pyo3_library_impl(ctx): files.append(ext) stub = None + pyi_files = [] if _stubs_enabled(ctx.attr.stubs, toolchain): stub = ctx.actions.declare_file(stub_relpath) @@ -146,6 +147,7 @@ def _py_pyo3_library_impl(ctx): arguments = [args], ) files.append(stub) + pyi_files.append(stub) providers = [ DefaultInfo( @@ -156,7 +158,9 @@ def _py_pyo3_library_impl(ctx): ).merge(ctx.attr.extension[DefaultInfo].default_runfiles), ), PyInfo( + direct_pyi_files = depset(direct = pyi_files), imports = _get_imports(ctx, ctx.attr.imports), + transitive_pyi_files = depset(direct = pyi_files), transitive_sources = depset(), ), coverage_common.instrumented_files_info(