Fix typing

healthjoy · Mar 19, 2024 · b877ac2 · b877ac2
1 parent 799c7f8
commit b877ac2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/async_firebase/base.py b/async_firebase/base.py
@@ -227,7 +227,7 @@ async def send_request(
         :return: HTTP response
         """
         url = join_url(self.BASE_URL, uri)
-        return await self._send_request(
+        return await self._send_request( # type: ignore
             url=url,
             response_handler=response_handler,
             json_payload=json_payload,
@@ -256,7 +256,7 @@ async def send_iid_request(
         url = join_url(self.IID_URL, uri)
         headers = headers or await self.prepare_headers()
         headers.update(self.IID_HEADERS)
-        return await self._send_request(
+        return await self._send_request( # type: ignore
             url=url,
             response_handler=response_handler,
             json_payload=json_payload,

diff --git a/async_firebase/messages.py b/async_firebase/messages.py
@@ -438,7 +438,7 @@ def __init__(
         self.exception = exception
         self._success_count = 0
         self._failure_count = 0
-        self._errors = []
+        self._errors: t.List[ErrorInfo] = []
 
         if resp:
             self._handle_response(resp)

diff --git a/async_firebase/utils.py b/async_firebase/utils.py
@@ -162,7 +162,7 @@ def serialize_mime_message(
     return fp.getvalue()
 
 
-FCMResponseType = t.TypeVar("FCMResponseType", FCMResponse, FCMBatchResponse)
+FCMResponseType = t.TypeVar("FCMResponseType", FCMResponse, FCMBatchResponse, TopicManagementResponse)
 
 
 class FCMResponseHandlerBase(ABC, t.Generic[FCMResponseType]):