Mypy fix

jinglemansweep · Nov 3, 2023 · 05423b3 · 05423b3
1 parent 2987c94
commit 05423b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wideboy/sprites/homeassistant/entity_row.py b/wideboy/sprites/homeassistant/entity_row.py
@@ -1,7 +1,7 @@
 import logging
 from pygame import Clock, Color, Event, Rect, Surface, SRCALPHA
 from jinja2 import Environment
-from typing import Optional, List, Set, Dict, Any
+from typing import Optional, List, Set, Dict, Any, Union
 from wideboy.constants import (
     EVENT_EPOCH_MINUTE,
     EVENT_EPOCH_SECOND,
@@ -78,7 +78,7 @@ def setup_watches(self) -> None:
         for watched_entities in self.entity_watches:
             self.entity_states[watched_entities] = ""
 
-    def parse_state_message(self, topic, payload) -> bool | None:
+    def parse_state_message(self, topic, payload) -> Union[bool, None]:
         topic_exploded = topic.split("/")
         if len(topic_exploded) < 3:
             return None