diff --git a/homeassistant/components/derivative/sensor.py b/homeassistant/components/derivative/sensor.py index a8cacc0e20d..adf91eb706b 100644 --- a/homeassistant/components/derivative/sensor.py +++ b/homeassistant/components/derivative/sensor.py @@ -8,11 +8,7 @@ from typing import TYPE_CHECKING import voluptuous as vol -from homeassistant.components.sensor import ( - PLATFORM_SCHEMA, - SensorEntity, - SensorStateClass, -) +from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( ATTR_UNIT_OF_MEASUREMENT, @@ -135,7 +131,6 @@ class DerivativeSensor(RestoreEntity, SensorEntity): _attr_icon = ICON _attr_should_poll = False - _attr_state_class = SensorStateClass.MEASUREMENT def __init__( self, diff --git a/tests/components/derivative/test_sensor.py b/tests/components/derivative/test_sensor.py index 9c2f68fd685..c1541812d1b 100644 --- a/tests/components/derivative/test_sensor.py +++ b/tests/components/derivative/test_sensor.py @@ -4,7 +4,6 @@ from math import sin import random from unittest.mock import patch -from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass from homeassistant.const import UnitOfPower, UnitOfTime from homeassistant.core import HomeAssistant from homeassistant.setup import async_setup_component @@ -79,7 +78,6 @@ async def setup_tests(hass, config, times, values, expected_state): assert state is not None assert round(float(state.state), config["sensor"]["round"]) == expected_state - assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT return state