Merge pull request #23835 from VadimLevin:dev/vlevin/fix-ast-nodes-required-usage-imports

fix: AST nodes required usage imports
pull/23838/head
Alexander Smorkalov 1 year ago committed by GitHub
commit fe4f5b539e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      modules/python/src2/typing_stubs_generation/generation.py
  2. 5
      modules/python/src2/typing_stubs_generation/nodes/type_node.py

@ -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

Loading…
Cancel
Save