From 2a6f4e1b805acd9380f04e3868d4b7e48b3db2d8 Mon Sep 17 00:00:00 2001 From: Kunal Gupta Date: Fri, 31 May 2024 01:35:47 +0530 Subject: [PATCH] Remove usage of deprecated validate_arguments Possible patch for issue: https://github.com/unionai-oss/pandera/issues/1663 --- pandera/decorators.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pandera/decorators.py b/pandera/decorators.py index fd044694d..36f85ac4c 100644 --- a/pandera/decorators.py +++ b/pandera/decorators.py @@ -23,7 +23,7 @@ import pandas as pd import wrapt -from pydantic import validate_arguments +from pydantic import validate_call from pandera import errors from pandera.api.base.error_handler import ErrorHandler @@ -566,7 +566,7 @@ def check_types( See the :ref:`User Guide ` for more. :param wrapped: the function to decorate. - :param with_pydantic: use ``pydantic.validate_arguments`` to validate + :param with_pydantic: use ``pydantic.validate_call`` to validate inputs. This function is still needed to validate function outputs. :param head: validate the first n rows. Rows overlapping with `tail` or `sample` are de-duplicated. @@ -832,7 +832,7 @@ async def _wrapper( kwargs: Dict[str, Any], ): if with_pydantic: - out = await validate_arguments(wrapped_)(*args, **kwargs) + out = await validate_call(wrapped_)(*args, **kwargs) else: validated_pos, validated_kwd = validate_inputs( instance, args, kwargs @@ -850,7 +850,7 @@ def _wrapper( kwargs: Dict[str, Any], ): if with_pydantic: - out = validate_arguments(wrapped_)(*args, **kwargs) + out = validate_call(wrapped_)(*args, **kwargs) else: validated_pos, validated_kwd = validate_inputs( instance, args, kwargs