fix: fix type error on BaseModel

codeforjapan · Oct 6, 2024 · 6c9300b · 6c9300b
1 parent 40ff893
commit 6c9300b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/birdxplorer_common/models.py b/common/birdxplorer_common/models.py
@@ -6,9 +6,9 @@
 from pydantic import BaseModel as PydanticBaseModel
 from pydantic import ConfigDict, GetCoreSchemaHandler, HttpUrl, TypeAdapter
 from pydantic.alias_generators import to_camel
+from pydantic.main import IncEx
 from pydantic_core import core_schema
 
-IncEx: TypeAlias = "set[int] | set[str] | dict[int, IncEx] | dict[str, IncEx] | None"
 StrT = TypeVar("StrT", bound="BaseString")
 IntT = TypeVar("IntT", bound="BaseInt")
 FloatT = TypeVar("FloatT", bound="BaseFloat")
@@ -467,8 +467,8 @@ def model_dump_json(
         self,
         *,
         indent: int | None = None,
-        include: IncEx = None,
-        exclude: IncEx = None,
+        include: IncEx | None = None,
+        exclude: IncEx | None = None,
         context: Dict[str, Any] | None = None,
         by_alias: bool = True,
         exclude_unset: bool = False,