fix: Fix typing info for sensors

eifinger · Apr 19, 2021 · e560461 · e560461
1 parent 3eac96c
commit e560461
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/custom_components/weenect/binary_sensor.py b/custom_components/weenect/binary_sensor.py
@@ -1,6 +1,6 @@
 """Binary_sensor platform for weenect."""
 import logging
-from typing import Dict, List
+from typing import Any, Dict, List
 
 from homeassistant.components.binary_sensor import BinarySensorEntity
 from homeassistant.core import callback
@@ -50,7 +50,7 @@ def __init__(
         self,
         coordinator: DataUpdateCoordinator,
         tracker_id: str,
-        sensor_type: Dict[str],
+        sensor_type: Dict[str, Any],
     ):
         super().__init__(coordinator, tracker_id)
         self._device_class = sensor_type["device_class"]

diff --git a/custom_components/weenect/sensor.py b/custom_components/weenect/sensor.py
@@ -1,6 +1,6 @@
 """Sensor platform for weenect."""
 import logging
-from typing import Dict, List
+from typing import Any, Dict, List
 
 from homeassistant.core import callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
@@ -47,7 +47,7 @@ def __init__(
         self,
         coordinator: DataUpdateCoordinator,
         tracker_id: str,
-        sensor_type: Dict[str],
+        sensor_type: Dict[str, Any],
     ):
         super().__init__(coordinator, tracker_id)
         self._device_class = sensor_type["device_class"]