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