From 2f279002b62de3c7dd889e13e4a760637f0927d3 Mon Sep 17 00:00:00 2001 From: Emmanuel Ferdman Date: Tue, 27 Jan 2026 01:44:27 +0200 Subject: [PATCH] [assertpy] Accept extensions with additional parameters Signed-off-by: Emmanuel Ferdman --- stubs/assertpy/assertpy/assertpy.pyi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stubs/assertpy/assertpy/assertpy.pyi b/stubs/assertpy/assertpy/assertpy.pyi index 5366825bcbd2..922a86dcf426 100644 --- a/stubs/assertpy/assertpy/assertpy.pyi +++ b/stubs/assertpy/assertpy/assertpy.pyi @@ -69,5 +69,5 @@ def assert_that(val: _V, description: str = "") -> AssertionBuilder[_V]: ... def assert_warn(val: _V, description: str = "", logger: logging.Logger | None = None) -> AssertionBuilder: ... def fail(msg: str = "") -> None: ... def soft_fail(msg: str = "") -> None: ... -def add_extension(func: Callable[[AssertionBuilder], AssertionBuilder]) -> None: ... -def remove_extension(func: Callable[[AssertionBuilder], AssertionBuilder]) -> None: ... +def add_extension(func: Callable[..., AssertionBuilder[Any]]) -> None: ... +def remove_extension(func: Callable[..., AssertionBuilder[Any]]) -> None: ...