Skip to content

Commit 4715617

Browse files
committed
updates logic to catch another annotation
1 parent f96d3f3 commit 4715617

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

scripts/microgenerator/generate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,7 @@ def visit_AnnAssign(self, node: ast.AnnAssign) -> None:
258258
# directly within the class body, not inside a method.
259259
elif isinstance(target, ast.Name) and not self._is_in_method:
260260
self._add_attribute(target.id, self._get_type_str(node.annotation))
261+
self._collect_types_from_node(node.annotation)
261262
self.generic_visit(node)
262263

263264

0 commit comments

Comments
 (0)