Add re.Pattern to allowed match types

Suggested in #436 (comment)
jd · Sep 24, 2024 · ed70a80 · ed70a80
1 parent 60358bb
commit ed70a80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tenacity/retry.py b/tenacity/retry.py
@@ -207,7 +207,7 @@ class retry_if_exception_message(retry_if_exception):
     def __init__(
         self,
         message: typing.Optional[str] = None,
-        match: typing.Optional[str] = None,
+        match: typing.Union[None, str, re.Pattern] = None,
     ) -> None:
         if message and match:
             raise TypeError(
@@ -242,7 +242,7 @@ class retry_if_not_exception_message(retry_if_exception_message):
     def __init__(
         self,
         message: typing.Optional[str] = None,
-        match: typing.Optional[str] = None,
+        match: typing.Union[None, str, re.Pattern] = None,
     ) -> None:
         super().__init__(message, match)
         # invert predicate