chore: add ruff PYI (#1026)

henryiii · web-flow · commit 0b1ca1cf1a0e · 2026-04-10T18:55:02.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -213,7 +213,7 @@ extend-select = [
   "PIE",    # flake8-pie
   "PT",     # flake8-pytest-style
  # "PTH",   # flake8-use-pathlib
- # "PYI",   # flake8-pyi
+  "PYI",    # flake8-pyi
  # "RET",   # flake8-return
   "RSE",    # flake8-raise
  # "SIM",   # flake8-simplify
diff --git a/src/build/_builder.py b/src/build/_builder.py
@@ -25,7 +25,7 @@
 import warnings
 import zipfile
 
-from typing import Any, TypeVar
+from typing import Any
 
 import pyproject_hooks
 
@@ -45,10 +45,12 @@
 if TYPE_CHECKING:
     from collections.abc import Iterator, Mapping, Sequence
 
-    from ._types import ConfigSettings, Distribution, StrPath, SubprocessRunner
-
+    if sys.version_info < (3, 11):
+        from typing_extensions import Self
+    else:
+        from typing import Self
 
-_TProjectBuilder = TypeVar('_TProjectBuilder', bound='ProjectBuilder')
+    from ._types import ConfigSettings, Distribution, StrPath, SubprocessRunner
 
 
 _DEFAULT_BACKEND = {
@@ -205,11 +207,11 @@ def __init__(
 
     @classmethod
     def from_isolated_env(
-        cls: type[_TProjectBuilder],
+        cls,
         env: env.IsolatedEnv,
         source_dir: StrPath,
         runner: SubprocessRunner = pyproject_hooks.default_subprocess_runner,
-    ) -> _TProjectBuilder:
+    ) -> Self:
         return cls(
             source_dir=source_dir,
             python_executable=env.python_executable,
diff --git a/src/build/_util.py b/src/build/_util.py
@@ -6,7 +6,8 @@
 TYPE_CHECKING = False
 
 if TYPE_CHECKING:
-    from collections.abc import Iterator, Set
+    from collections.abc import Iterator
+    from collections.abc import Set as AbstractSet
 
 
 _WHEEL_FILENAME_REGEX = re.compile(
@@ -17,7 +18,7 @@
 
 
 def check_dependency(
-    req_string: str, ancestral_req_strings: tuple[str, ...] = (), parent_extras: Set[str] = frozenset()
+    req_string: str, ancestral_req_strings: tuple[str, ...] = (), parent_extras: AbstractSet[str] = frozenset()
 ) -> Iterator[tuple[str, ...]]:
     """
     Verify that a dependency and all of its dependencies are met.
diff --git a/src/build/env.py b/src/build/env.py
@@ -46,6 +46,11 @@
 
     from ._compat.importlib import metadata as importlib_metadata
 
+    if sys.version_info < (3, 11):
+        from typing_extensions import Self
+    else:
+        from typing import Self
+
 
 Installer = typing.Literal['pip', 'uv']
 
@@ -103,7 +108,7 @@ def __init__(
     ) -> None:
         self.installer: Installer = installer
 
-    def __enter__(self) -> DefaultIsolatedEnv:
+    def __enter__(self) -> Self:
         try:
             path = tempfile.mkdtemp(prefix='build-env-')
             # Call ``realpath`` to prevent spurious warning from being emitted
