ruff fixes

Author: dirixmjm

custom_components/plugwise_usb/coordinator.py

@@ -1,9 +1,10 @@
 """DataUpdateCoordinator for Plugwise USB-Stick."""
 
 from collections import Counter
+from collections.abc import Callable
 from datetime import timedelta
 import logging
-from typing import Any, Callable
+from typing import Any
 
 from plugwise_usb.api import PUSHING_FEATURES, NodeFeature, PlugwiseNode
 from plugwise_usb.exceptions import NodeError, NodeTimeout, StickError, StickTimeout

custom_components/plugwise_usb/event.py

@@ -9,11 +9,11 @@
 from plugwise_usb.api import NodeEvent, NodeFeature
 
 from homeassistant.components.event import (
-    EventDeviceClass, 
-    EventEntity, 
+    EventDeviceClass,
+    EventEntity,
     EventEntityDescription,
 )
-from homeassistant.const import EntityCategory, Platform, UnitOfTime
+from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
@@ -141,8 +141,8 @@ def _handle_coordinator_update(self) -> None:
             )
             return
         # SWITCH logic
-        state_value = getattr(data, "state" )
-        group_value = getattr(data, "group" )
+        state_value = data.state
+        group_value = data.group
         match self.entity_description.key:
             case "button_press_i_group_1":
                 if state_value is True and group_value == 1: