Use new UnitOf enums

kukulich · Jan 5, 2023 · 29bd0b2 · 29bd0b2
1 parent e00a23e
commit 29bd0b2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/custom_components/jablotron100/sensor.py b/custom_components/jablotron100/sensor.py
@@ -5,9 +5,9 @@
 	SensorStateClass,
 )
 from homeassistant.const import (
-	ELECTRIC_CURRENT_MILLIAMPERE,
-	ELECTRIC_POTENTIAL_VOLT,
 	PERCENTAGE,
+	UnitOfElectricCurrent,
+	UnitOfElectricPotential,
 	UnitOfTemperature,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -51,14 +51,14 @@
 	EntityType.VOLTAGE: SensorEntityDescription(
 		key=EntityType.VOLTAGE,
 		state_class=SensorStateClass.MEASUREMENT,
-		native_unit_of_measurement=ELECTRIC_POTENTIAL_VOLT,
+		native_unit_of_measurement=UnitOfElectricPotential.VOLT,
 		device_class=SensorDeviceClass.VOLTAGE,
 		entity_category=EntityCategory.DIAGNOSTIC,
 	),
 	EntityType.CURRENT: SensorEntityDescription(
 		key=EntityType.CURRENT,
 		state_class=SensorStateClass.MEASUREMENT,
-		native_unit_of_measurement=ELECTRIC_CURRENT_MILLIAMPERE,
+		native_unit_of_measurement=UnitOfElectricCurrent.MILLIAMPERE,
 		device_class=SensorDeviceClass.CURRENT,
 		entity_category=EntityCategory.DIAGNOSTIC,
 	),