Use better return type for inspect.waktree and `inspect.getclasstre…

…e` (#13129)
python · Dec 16, 2024 · 54e1c6a · 54e1c6a
1 parent 7cae61f
commit 54e1c6a
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/stdlib/inspect.pyi b/stdlib/inspect.pyi
@@ -421,11 +421,10 @@ class BoundArguments:
 # Classes and functions
 #
 
-# TODO: The actual return type should be list[_ClassTreeItem] but mypy doesn't
-# seem to be supporting this at the moment:
-# _ClassTreeItem = list[_ClassTreeItem] | Tuple[type, Tuple[type, ...]]
-def getclasstree(classes: list[type], unique: bool = False) -> list[Any]: ...
-def walktree(classes: list[type], children: Mapping[type[Any], list[type]], parent: type[Any] | None) -> list[Any]: ...
+_ClassTreeItem: TypeAlias = list[tuple[type, ...]] | list[_ClassTreeItem]
+
+def getclasstree(classes: list[type], unique: bool = False) -> _ClassTreeItem: ...
+def walktree(classes: list[type], children: Mapping[type[Any], list[type]], parent: type[Any] | None) -> _ClassTreeItem: ...
 
 class Arguments(NamedTuple):
     args: list[str]