refactor: Use annotation getter for base classes

mkdocstrings · Dec 27, 2021 · 8b1a7ed · 8b1a7ed
1 parent 6edfc9f
commit 8b1a7ed
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/griffe/agents/visitor.py b/src/griffe/agents/visitor.py
@@ -20,7 +20,6 @@
 from griffe.agents.nodes import (
     ASTNode,
     get_annotation,
-    get_baseclass,
     get_docstring,
     get_instance_names,
     get_names,
@@ -219,15 +218,15 @@ def visit_classdef(self, node: ast.ClassDef) -> None:
         bases = []
         if node.bases:
             for base in node.bases:
-                bases.append(get_baseclass(base, self.current))
+                bases.append(get_annotation(base, self.current))
 
         class_ = Class(
             name=node.name,
             lineno=lineno,
             endlineno=node.end_lineno,  # type: ignore[attr-defined]
             docstring=self._get_docstring(node),
             decorators=decorators,
-            bases=bases,
+            bases=bases,  # type: ignore[arg-type]
         )
         self.current[node.name] = class_
         self.current = class_

diff --git a/src/griffe/dataclasses.py b/src/griffe/dataclasses.py
@@ -960,7 +960,7 @@ class Class(Object):
     def __init__(
         self,
         *args: Any,
-        bases: list[Name | Expression] | None = None,
+        bases: list[Name | Expression | str] | None = None,
         decorators: list[Decorator] | None = None,
         **kwargs: Any,
     ) -> None:
@@ -973,7 +973,7 @@ def __init__(
             **kwargs: See [`griffe.dataclasses.Object`][].
         """
         super().__init__(*args, **kwargs)
-        self.bases: list[Name | Expression] = bases or []
+        self.bases: list[Name | Expression | str] = bases or []
         self.decorators: list[Decorator] = decorators or []
 
     def as_dict(self, **kwargs: Any) -> dict[str, Any]:  # type: ignore[override]