refactor: Rename visitor method

mkdocstrings · Nov 13, 2021 · 3e0c43c · 3e0c43c
1 parent ac86f17
commit 3e0c43c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/griffe/visitor.py b/src/griffe/visitor.py
@@ -612,7 +612,7 @@ def visit_ImportFrom(self, node) -> None:
  # self.scope[self.path][alias.asname or alias.name] = f"{node.module}.{alias.name}"
  self.generic_visit(node)
 
- def handle_data(self, node, annotation: str | None = None): # noqa: WPS231
+ def handle_attribute(self, node, annotation: str | Name | Expression | None = None): # noqa: WPS231
  parent = self.current
  labels = set()
 
@@ -657,7 +657,7 @@ def handle_data(self, node, annotation: str | None = None): # noqa: WPS231
  parent[name] = attribute # type: ignore
 
  def visit_Assign(self, node) -> None:
- self.handle_data(node)
+ self.handle_attribute(node)
 
  def visit_AnnAssign(self, node) -> None:
- self.handle_data(node, _get_annotation(node))
+ self.handle_attribute(node, _get_annotation(node.annotation, parent=self.current))