fix: 🚑 invalid exc_response return type hint

lucas-labs · Jan 26, 2024 · b6c100a · b6c100a
1 parent 8079101
commit b6c100a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pest/exceptions/__init__.py b/pest/exceptions/__init__.py
@@ -23,6 +23,7 @@
     UnauthorizedException,
     UnprocessableEntityException,
     UnsupportedMediaTypeException,
+    exc_response,
 )
 
 __all__ = [
@@ -50,4 +51,5 @@
     'GatewayTimeoutException',
     'HttpVersionNotSupportedException',
     'PestException',
+    'exc_response',
 ]
diff --git a/pest/exceptions/http/http.py b/pest/exceptions/http/http.py
@@ -25,7 +25,7 @@ def example(code: int) -> 'ExceptionResponse':
         )
 
 
-def exc_response(*codes: int) -> Dict[int, Dict[str, Any]]:
+def exc_response(*codes: int) -> Dict[Union[int, str], Dict[str, Any]]:
     """🐀 ⇝ Generate a dict of error responses for the given status codes for use in OpenAPI docs"""
 
     responses = {}