From 1ad4cb40dfbad8cc5bcb7183d3a68e8057196a2e Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 13 Dec 2022 10:18:35 +0100
Subject: [PATCH] Use DATA_SIZE device class in netgear lte (#83908)

---
 homeassistant/components/netgear_lte/sensor.py       | 6 ++++--
 homeassistant/components/netgear_lte/sensor_types.py | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/netgear_lte/sensor.py b/homeassistant/components/netgear_lte/sensor.py
index c27c4f43920..49942306da2 100644
--- a/homeassistant/components/netgear_lte/sensor.py
+++ b/homeassistant/components/netgear_lte/sensor.py
@@ -1,7 +1,7 @@
 """Support for Netgear LTE sensors."""
 from __future__ import annotations
 
-from homeassistant.components.sensor import SensorEntity
+from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import PlatformNotReady
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -73,8 +73,10 @@ class SMSTotalSensor(LTESensor):
 class UsageSensor(LTESensor):
     """Data usage sensor entity."""
 
+    _attr_device_class = SensorDeviceClass.DATA_SIZE
+
     @property
-    def native_value(self):
+    def native_value(self) -> float:
         """Return the state of the sensor."""
         return round(self.modem_data.data.usage / 1024**2, 1)
 
diff --git a/homeassistant/components/netgear_lte/sensor_types.py b/homeassistant/components/netgear_lte/sensor_types.py
index 45cfb873ef8..01aa267e953 100644
--- a/homeassistant/components/netgear_lte/sensor_types.py
+++ b/homeassistant/components/netgear_lte/sensor_types.py
@@ -2,9 +2,9 @@
 
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass
 from homeassistant.const import (
-    DATA_MEBIBYTES,
     PERCENTAGE,
     SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+    UnitOfInformation,
 )
 
 SENSOR_SMS = "sms"
@@ -14,7 +14,7 @@ SENSOR_USAGE = "usage"
 SENSOR_UNITS = {
     SENSOR_SMS: "unread",
     SENSOR_SMS_TOTAL: "messages",
-    SENSOR_USAGE: DATA_MEBIBYTES,
+    SENSOR_USAGE: UnitOfInformation.MEBIBYTES,
     "radio_quality": PERCENTAGE,
     "rx_level": SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
     "tx_level": SIGNAL_STRENGTH_DECIBELS_MILLIWATT,