Skip to content

Commit ded6e7a

Browse files
committed
Need to set in init.
1 parent 8555f3e commit ded6e7a

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

zhaquirks/tuya/mcu/__init__.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
import datetime
66
from typing import Any, Optional, Union
77

8+
from zigpy.quirks import _DEVICE_REGISTRY
9+
from zigpy.quirks.registry import DeviceRegistry
810
from zigpy.quirks.v2 import QuirkBuilder, QuirksV2RegistryEntry
911
import zigpy.types as t
1012
from zigpy.zcl import foundation
@@ -730,9 +732,13 @@ class TuyaLevelControlManufCluster(TuyaMCUCluster):
730732
class TuyaQuirkBuilder(QuirkBuilder):
731733
"""Tuya QuirkBuilder."""
732734

733-
tuya_attributes = TuyaMCUCluster.attributes.copy()
734-
tuya_data_point_handlers: dict[int, str] = {}
735-
tuya_dp_to_attribute: dict[int, DPToAttributeMapping] = {}
735+
def __init__(
736+
self, manufacturer: str, model: str, registry: DeviceRegistry = _DEVICE_REGISTRY
737+
) -> None:
738+
"""Init the TuyaQuirkBuilder."""
739+
self.tuya_data_point_handlers: dict[int, str] = {}
740+
self.tuya_dp_to_attribute: dict[int, DPToAttributeMapping] = {}
741+
super().__init__(manufacturer, model, registry)
736742

737743
def add_tuya_dp(
738744
self,

0 commit comments

Comments
 (0)