From 01d0da49ffdc598d4b3d54d34a276a71679ef19c Mon Sep 17 00:00:00 2001 From: TheJulianJES Date: Fri, 11 Oct 2024 06:10:55 +0200 Subject: [PATCH] Fix `quirk_applied` check only working for v1 quirks (#232) Check for instance of `BaseCustomDevice` `CustomDeviceV2` no longer extends `CustomDevice`, only `BaseCustomDevice`, so we need to check that here. --- zha/zigbee/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zha/zigbee/device.py b/zha/zigbee/device.py index 9f235fcc..c7c3f2af 100644 --- a/zha/zigbee/device.py +++ b/zha/zigbee/device.py @@ -207,7 +207,7 @@ def __init__( self._gateway: Gateway = _gateway self._zigpy_device: ZigpyDevice = zigpy_device self.quirk_applied: bool = isinstance( - self._zigpy_device, zigpy.quirks.CustomDevice + self._zigpy_device, zigpy.quirks.BaseCustomDevice ) self.quirk_class: str = ( f"{self._zigpy_device.__class__.__module__}."