From 7f3ec0016092743c1d6738e61bed560614d4def8 Mon Sep 17 00:00:00 2001
From: Emmanuel Ferdman <emmanuelferdman@gmail.com>
Date: Tue, 27 Jan 2026 08:23:56 +0200
Subject: [PATCH] [assertpy] Accept extensions with additional parameters
 (#15329)

---
 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 5366825bc..922a86dcf 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: ...