From 86b688b35efdbecbec7b07de948a5308567480da Mon Sep 17 00:00:00 2001 From: Ofek Weiss Date: Wed, 12 Mar 2025 12:13:59 +0200 Subject: [PATCH] fixed typing for mapped messaging integration --- elementary/messages/messaging_integrations/mapped.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/elementary/messages/messaging_integrations/mapped.py b/elementary/messages/messaging_integrations/mapped.py index 22e3604b7..cd0e247e3 100644 --- a/elementary/messages/messaging_integrations/mapped.py +++ b/elementary/messages/messaging_integrations/mapped.py @@ -1,4 +1,4 @@ -from typing import Dict +from typing import Generic, Mapping from elementary.messages.message_body import MessageBody from elementary.messages.messaging_integrations.base_messaging_integration import ( @@ -11,9 +11,12 @@ ) -class MappedMessagingIntegration(BaseMessagingIntegration[str, MessageContextType]): +class MappedMessagingIntegration( + Generic[MessageContextType], + BaseMessagingIntegration[str, MessageContextType], +): def __init__( - self, mapping: Dict[str, BaseMessagingIntegration[None, MessageContextType]] + self, mapping: Mapping[str, BaseMessagingIntegration[None, MessageContextType]] ): self._mapping = mapping