style(typing): fix typing for defaultdicts

xaviml · Apr 28, 2021 · 529786d · 529786d
1 parent 3ee7e4e
commit 529786d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/apps/controllerx/cx_core/controller.py b/apps/controllerx/cx_core/controller.py
@@ -11,6 +11,7 @@
     Awaitable,
     Callable,
     Counter,
+    DefaultDict,
     Dict,
     List,
     Optional,
@@ -85,15 +86,17 @@ class Controller(Hass, Mqtt):
 
     integration: Integration
     actions_mapping: ActionsMapping
-    action_handles: Dict[ActionEvent, Optional[Future]]
+    action_handles: DefaultDict[ActionEvent, Optional["Future[None]"]]
     action_delay_handles: Dict[ActionEvent, Optional[float]]
     multiple_click_actions: Set[ActionEvent]
     action_delay: Dict[ActionEvent, int]
     action_delta: Dict[ActionEvent, int]
     action_times: Dict[str, float]
     multiple_click_action_times: Dict[str, float]
     click_counter: Counter[ActionEvent]
-    multiple_click_action_delay_tasks: Dict[ActionEvent, Optional[Future]]
+    multiple_click_action_delay_tasks: DefaultDict[
+        ActionEvent, Optional["Future[None]"]
+    ]
     multiple_click_delay: int
 
     async def initialize(self) -> None: