Update sensor.py

custom-components · Jun 11, 2024 · 8cf48cc · 8cf48cc
1 parent 5778dcd
commit 8cf48cc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/custom_components/trakt/sensor.py b/custom_components/trakt/sensor.py
@@ -1,5 +1,7 @@
 """Sensor platform for Trakt."""
 
+from typing import Any
+
 from homeassistant.components.sensor import SensorEntity
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_CLIENT_ID, CONF_NAME
@@ -42,6 +44,6 @@ def state(self) -> int:
         return len(self.coordinator.tv_shows)
 
     @property
-    def extra_state_attributes(self) -> dict[str, str]:
+    def extra_state_attributes(self) -> dict[str, Any]:
         """Return the state attributes of the sensor."""
         return {"data": self.coordinator.data}