From f10b4fd653672c39a3f6e0ab9a039fd565bbe3fa Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 27 Nov 2024 19:56:46 +0000 Subject: [PATCH] Apply pre-commit auto fixes --- zhaquirks/tuya/builder/__init__.py | 15 ++++++++------- zhaquirks/tuya/ts0601_valve.py | 19 ++++++++++--------- 2 files changed, 18 insertions(+), 16 deletions(-) diff --git a/zhaquirks/tuya/builder/__init__.py b/zhaquirks/tuya/builder/__init__.py index 1cdbf93981..2c0159d4b0 100644 --- a/zhaquirks/tuya/builder/__init__.py +++ b/zhaquirks/tuya/builder/__init__.py @@ -4,13 +4,6 @@ from enum import Enum from typing import Any, Optional -from zhaquirks.tuya import ( - TUYA_CLUSTER_ID, - PowerConfiguration, - TuyaLocalCluster, - TuyaPowerConfigurationCluster2AAA, -) -from zhaquirks.tuya.mcu import DPToAttributeMapping, TuyaMCUCluster, TuyaOnOffNM from zigpy.quirks import _DEVICE_REGISTRY from zigpy.quirks.registry import DeviceRegistry from zigpy.quirks.v2 import QuirkBuilder, QuirksV2RegistryEntry @@ -27,6 +20,14 @@ ) from zigpy.zcl.clusters.smartenergy import Metering +from zhaquirks.tuya import ( + TUYA_CLUSTER_ID, + PowerConfiguration, + TuyaLocalCluster, + TuyaPowerConfigurationCluster2AAA, +) +from zhaquirks.tuya.mcu import DPToAttributeMapping, TuyaMCUCluster, TuyaOnOffNM + class TuyaRelativeHumidity(RelativeHumidity, TuyaLocalCluster): """Tuya local RelativeHumidity cluster.""" diff --git a/zhaquirks/tuya/ts0601_valve.py b/zhaquirks/tuya/ts0601_valve.py index 6097f44031..834c11f967 100644 --- a/zhaquirks/tuya/ts0601_valve.py +++ b/zhaquirks/tuya/ts0601_valve.py @@ -2,6 +2,16 @@ from datetime import datetime, timedelta, timezone +from zigpy.profiles import zha +from zigpy.quirks import CustomDevice +from zigpy.quirks.v2 import EntityPlatform, EntityType +from zigpy.quirks.v2.homeassistant import UnitOfTime +from zigpy.quirks.v2.homeassistant.sensor import SensorDeviceClass, SensorStateClass +import zigpy.types as t +from zigpy.zcl import foundation +from zigpy.zcl.clusters.general import Basic, Groups, Identify, OnOff, Ota, Scenes, Time +from zigpy.zcl.clusters.smartenergy import Metering + from zhaquirks import DoublingPowerConfigurationCluster from zhaquirks.const import ( DEVICE_TYPE, @@ -25,15 +35,6 @@ TuyaOnOff, TuyaPowerConfigurationCluster, ) -from zigpy.profiles import zha -from zigpy.quirks import CustomDevice -from zigpy.quirks.v2 import EntityPlatform, EntityType -from zigpy.quirks.v2.homeassistant import UnitOfTime -from zigpy.quirks.v2.homeassistant.sensor import SensorDeviceClass, SensorStateClass -import zigpy.types as t -from zigpy.zcl import foundation -from zigpy.zcl.clusters.general import Basic, Groups, Identify, OnOff, Ota, Scenes, Time -from zigpy.zcl.clusters.smartenergy import Metering class TuyaValveWaterConsumed(Metering, TuyaLocalCluster):