提交 06b40aef 编写于 作者: V Vadim Levin

fix: AST nodes required usage imports

上级 726ba021
......@@ -759,8 +759,6 @@ def _generate_typing_module(root: NamespaceNode, output_path: Path) -> None:
output_stream.write(f' "{alias_name}",\n')
output_stream.write("]\n\n")
# HACK: force add cv2.mat_wrapper import to handle MatLike alias
required_imports.add("import cv2.mat_wrapper")
_write_required_imports(required_imports, output_stream)
# Add type checking time definitions as generated __init__.py content
......
......@@ -561,13 +561,14 @@ class ContainerTypeNode(AggregatedTypeNode):
@property
def required_definition_imports(self) -> Generator[str, None, None]:
yield "import typing"
return super().required_definition_imports
yield from super().required_definition_imports
@property
def required_usage_imports(self) -> Generator[str, None, None]:
if TypeNode.compatible_to_runtime_usage:
yield "import typing"
return super().required_usage_imports
yield from super().required_usage_imports
@abc.abstractproperty
def type_format(self) -> str:
pass
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册