feat: make pypi-generated targets include pyi files

Author: rickeylev

CHANGELOG.md

@@ -67,6 +67,8 @@ Unreleased changes template.
   `python_version` attribute is still used to specify the Python version.
 * (pypi) Updated versions of packages: `pip` to 24.3.1 and
   `packaging` to 24.2.
+* (pypi) For pypi-generated targets, `*.pyi` files are included in the
+  `pyi_srcs` attribute instead of the `data` attribute.
 
 {#v0-0-0-deprecations}
 #### Deprecations
@@ -120,6 +122,7 @@ Unreleased changes template.
   only dependencies added. See {obj}`py_library.pyi_srcs` and
   `py_library.pyi_deps` (and the same named attributes for `py_binary` and
   `py_test`).
+* (pypi) pypi-generated targets set `pyi_srcs` to include `*.pyi` files.
 * (providers) {obj}`PyInfo` has new fields to aid static analysis tools:
   {obj}`direct_original_sources`, {obj}`direct_pyi_files`,
   {obj}`transitive_original_sources`, {obj}`transitive_pyi_files`.

python/private/pypi/whl_library_targets.bzl

@@ -226,6 +226,7 @@ def whl_library_targets(
             "**/*.py",
             "**/*.pyc",
             "**/*.pyc.*",  # During pyc creation, temp files named *.pyc.NNNN are created
+            "**/*.pyi",
             # RECORD is known to contain sha256 checksums of files which might include the checksums
             # of generated files produced when wheels are installed. The file is ignored to avoid
             # Bazel caching issues.
@@ -244,6 +245,9 @@ def whl_library_targets(
                 # pure-Python code, e.g. pymssql, which is written in Cython.
                 allow_empty = True,
             ),
+            pyi_srcs = native.glob(
+                ["site-packages/**/*.pyi"],
+            ),
             data = data + native.glob(
                 ["site-packages/**/*"],
                 exclude = _data_exclude,