From 07d0b6f726597bda14a4fec1183d82e72819dfbc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Hjelseth=20H=C3=B8yer?= <mail@dahoiv.net>
Date: Wed, 8 Dec 2021 22:08:05 +0100
Subject: [PATCH] Use new EntityCategory in rfxtrx (#61295)

---
 homeassistant/components/rfxtrx/sensor.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/rfxtrx/sensor.py b/homeassistant/components/rfxtrx/sensor.py
index d4d37bcd07be..c13e499bbf03 100644
--- a/homeassistant/components/rfxtrx/sensor.py
+++ b/homeassistant/components/rfxtrx/sensor.py
@@ -19,7 +19,6 @@ from homeassistant.const import (
     ELECTRIC_CURRENT_AMPERE,
     ELECTRIC_POTENTIAL_VOLT,
     ENERGY_KILO_WATT_HOUR,
-    ENTITY_CATEGORY_DIAGNOSTIC,
     LENGTH_MILLIMETERS,
     PERCENTAGE,
     POWER_WATT,
@@ -31,6 +30,7 @@ from homeassistant.const import (
     UV_INDEX,
 )
 from homeassistant.core import callback
+from homeassistant.helpers.entity import EntityCategory
 
 from . import (
     CONF_DATA_BITS,
@@ -78,7 +78,7 @@ SENSOR_TYPES = (
         state_class=SensorStateClass.MEASUREMENT,
         native_unit_of_measurement=PERCENTAGE,
         convert=_battery_convert,
-        entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
+        entity_category=EntityCategory.DIAGNOSTIC,
     ),
     RfxtrxSensorEntityDescription(
         key="Current",
@@ -122,7 +122,7 @@ SENSOR_TYPES = (
         state_class=SensorStateClass.MEASUREMENT,
         native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
         convert=_rssi_convert,
-        entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
+        entity_category=EntityCategory.DIAGNOSTIC,
     ),
     RfxtrxSensorEntityDescription(
         key="Temperature",