Typing

andrew-codechimp · Oct 30, 2024 · 9dc4e6b · 9dc4e6b
1 parent 80d9a2a
commit 9dc4e6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/hive_local_thermostat/entity.py b/custom_components/hive_local_thermostat/entity.py
@@ -4,7 +4,7 @@
 
 import abc
 from dataclasses import dataclass
-from typing import Any
+from typing import Any, cast
 
 from homeassistant.helpers.entity import DeviceInfo, Entity, EntityDescription
 
@@ -52,4 +52,4 @@ def get_entity_value(self, entity_key: str, default: float) -> float:
         if self.entity_description.entry_id not in self.hass.data[DOMAIN]:
             return default
 
-        return self.hass.data[DOMAIN][self.entity_description.entry_id].get(entity_key, default)
+        return cast(float, self.hass.data[DOMAIN][self.entity_description.entry_id].get(entity_key, default))