chore: update types

pilotak · Dec 21, 2024 · 7b39fc6 · 7b39fc6
1 parent c1daa17
commit 7b39fc6
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 12 deletions.
diff --git a/custom_components/mcp2221/binary_sensor.py b/custom_components/mcp2221/binary_sensor.py
@@ -1,5 +1,6 @@
 """MCP2221 binary sensor"""
 
+from asyncio import Lock
 from datetime import datetime, timedelta
 from homeassistant.components.binary_sensor import (
     BinarySensorDeviceClass,
@@ -56,7 +57,7 @@ async def async_setup_platform(
         }
 
         # get MCP2221 instance
-        device_instance: MCP2221.MCP2221() | None = hass.data[DOMAIN].get(
+        device_instance = hass.data[DOMAIN].get(
             binary_sensor.get(CONF_DEVICE_ID))
 
         if not isinstance(device_instance["device"], MCP2221.MCP2221):
@@ -84,15 +85,15 @@ class MCP2221BinarySensor(ManualTriggerEntity, BinarySensorEntity):
     def __init__(
         self,
         config: ConfigType,
-        device: MCP2221,
+        device,
         pin: int,
         inverted: bool,
         interval: timedelta,
     ) -> None:
         """Initialize the binary sensor."""
         super().__init__(self.hass, config)
-        self._device = device["device"]
-        self._lock = device["lock"]
+        self._device: MCP2221.MCP2221 = device["device"]
+        self._lock: Lock = device["lock"]
         self._pin = pin
         self._inverted = inverted
         self._scan_interval = interval

diff --git a/custom_components/mcp2221/sensor.py b/custom_components/mcp2221/sensor.py
@@ -1,5 +1,6 @@
 """MCP2221 sensor"""
 
+from asyncio import Lock
 from datetime import datetime, timedelta
 from typing import Any
 
@@ -68,7 +69,7 @@ async def async_setup_platform(
             trigger_entity_config[key] = sensor.get(key)
 
         # get MCP2221 instance
-        device_instance: MCP2221.MCP2221() | None = hass.data[DOMAIN].get(
+        device_instance = hass.data[DOMAIN].get(
             sensor.get(CONF_DEVICE_ID))
 
         if not isinstance(device_instance["device"], MCP2221.MCP2221):
@@ -113,14 +114,14 @@ def __init__(
         hass: HomeAssistant,
         config: ConfigType,
         value_template: Template | None,
-        device: MCP2221,
+        device,
         pin: int,
         scan_interval: timedelta,
     ) -> None:
         """Initialize the sensor."""
         ManualTriggerSensorEntity.__init__(self, hass, config)
-        self._device = device["device"]
-        self._lock = device["lock"]
+        self._device: MCP2221.MCP2221 = device["device"]
+        self._lock: Lock = device["lock"]
         self._pin = pin
         self._scan_interval = scan_interval
         self._value_template = value_template

diff --git a/custom_components/mcp2221/switch.py b/custom_components/mcp2221/switch.py
@@ -1,5 +1,6 @@
 """MCP2221 switch"""
 
+from asyncio import Lock
 from homeassistant.components.switch import SwitchEntity
 from homeassistant.const import (
     CONF_PIN,
@@ -43,7 +44,7 @@ async def async_setup_platform(
 
         # get MCP2221 instance
 
-        device_instance: MCP2221.MCP2221() | None = hass.data[DOMAIN].get(
+        device_instance = hass.data[DOMAIN].get(
             switch.get(CONF_DEVICE_ID))
 
         if not isinstance(device_instance["device"], MCP2221.MCP2221):
@@ -67,13 +68,13 @@ class MCP2221Switch(ManualTriggerEntity, SwitchEntity):
     def __init__(
         self,
         config: ConfigType,
-        device: MCP2221,
+        device,
         pin: int
     ) -> None:
         """Initialize the switch."""
         super().__init__(self.hass, config)
-        self._device = device["device"]
-        self._lock = device["lock"]
+        self._device: MCP2221.MCP2221 = device["device"]
+        self._lock: Lock = device["lock"]
         self._pin = pin
 
         # init GP