Merge branch 'main' into py13

drivendataorg · Oct 18, 2024 · 92721b1 · 92721b1
2 parents 26e9477 + cac87b9
commit 92721b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/erdantic/core.py b/erdantic/core.py
@@ -14,7 +14,7 @@
     from typing_extensions import Self
 
 import pydantic
-import pygraphviz as pgv  # type: ignore [import-not-found]
+import pygraphviz as pgv  # type: ignore [import-untyped, import-not-found]
 from sortedcontainers_pydantic import SortedDict
 from typenames import REMOVE_ALL_MODULES, typenames
 

diff --git a/erdantic/plugins/dataclasses.py b/erdantic/plugins/dataclasses.py
@@ -1,7 +1,7 @@
 import dataclasses
 import re
 import sys
-from typing import TYPE_CHECKING, Any, List, Type, get_type_hints
+from typing import TYPE_CHECKING, Any, List, Type, cast, get_type_hints
 
 if sys.version_info >= (3, 9):
     # include_extras was added in Python 3.9
@@ -69,7 +69,7 @@ def get_fields_from_dataclass(model: DataclassType) -> List[FieldInfo]:
         FieldInfo.from_raw_type(
             model_full_name=FullyQualifiedName.from_object(model),
             name=f.name,
-            raw_type=f.type,
+            raw_type=cast(type, f.type),  # cast narrows type for typechecking
         )
         for f in dataclasses.fields(model)
     ]