Fix type stubs for annotated versions

surenkov · Mar 26, 2024 · 871af26 · 871af26
1 parent 59203db
commit 871af26
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/django_pydantic_field/fields.pyi b/django_pydantic_field/fields.pyi
@@ -87,7 +87,7 @@ def SchemaField(
 ) -> ST | None: ...
 @ty.overload
 def SchemaField(
-    schema: te.Annotated[type[ST | None], ...] = ...,
+    schema: te.Annotated[ty.Type[ST | None], ...] = ...,
     config: ConfigType = ...,
     default: OptSchemaT | ty.Callable[[], OptSchemaT] | BaseExpression = ...,
     *args,
@@ -105,7 +105,7 @@ def SchemaField(
 ) -> ST: ...
 @ty.overload
 def SchemaField(
-    schema: te.Annotated[ST, ...] = ...,
+    schema: te.Annotated[ty.Type[ST], ...] = ...,
     config: ConfigType = ...,
     default: SchemaT | ty.Callable[[], SchemaT] | BaseExpression = ...,
     *args,

diff --git a/django_pydantic_field/v2/fields.py b/django_pydantic_field/v2/fields.py
@@ -225,7 +225,7 @@ def __call__(self, col: Col | None = None, *args, **kwargs) -> Transform | None:
 
 @ty.overload
 def SchemaField(
-    schema: ty.Annotated[types.ST | None, ...] = ...,
+    schema: ty.Annotated[type[types.ST | None], ...] = ...,
     config: pydantic.ConfigDict = ...,
     default: types.SchemaT | ty.Callable[[], types.SchemaT | None] | BaseExpression | None = ...,
     *args,