fix(sensor): inherit from SensorEntity

closes #1633
alandtse · Jun 26, 2022 · c28b8ef · c28b8ef
1 parent 22a6eb1
commit c28b8ef
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/alexa_media/sensor.py b/custom_components/alexa_media/sensor.py
@@ -10,6 +10,7 @@
 import logging
 from typing import Callable, List, Optional, Text  # noqa pylint: disable=unused-import
 
+from homeassistant.components.sensor import SensorEntity
 from homeassistant.const import (
     DEVICE_CLASS_TIMESTAMP,
     STATE_UNAVAILABLE,
@@ -18,7 +19,6 @@
 )
 from homeassistant.exceptions import ConfigEntryNotReady, NoEntitySpecifiedError
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
-from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.event import async_track_point_in_utc_time
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.util import dt
@@ -182,7 +182,7 @@ def lookup_device_info(account_dict, device_serial):
     return None
 
 
-class TemperatureSensor(CoordinatorEntity):
+class TemperatureSensor(SensorEntity, CoordinatorEntity):
     """A temperature sensor reported by an Echo."""
 
     def __init__(self, coordinator, entity_id, name, media_player_device_id):
@@ -222,7 +222,7 @@ def unique_id(self):
         return self.alexa_entity_id + "_temperature"
 
 
-class AlexaMediaNotificationSensor(Entity):
+class AlexaMediaNotificationSensor(SensorEntity):
     """Representation of Alexa Media sensors."""
 
     def __init__(