fix: type-error on model_validator in wrap mode (#7496)

pydantic · Sep 19, 2023 · f468466 · f468466
1 parent 22367fa
commit f468466
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pydantic/functional_validators.py b/pydantic/functional_validators.py
@@ -432,7 +432,9 @@ def __call__(  # noqa: D102
 def model_validator(
     *,
     mode: Literal['wrap'],
-) -> Callable[[_AnyModelWrapValidator], _decorators.PydanticDescriptorProxy[_decorators.ModelValidatorDecoratorInfo]]:
+) -> Callable[
+    [_AnyModelWrapValidator[_ModelType]], _decorators.PydanticDescriptorProxy[_decorators.ModelValidatorDecoratorInfo]
+]:
     ...