From bfb5dd1cef8196b5c79fa85c255a152a7e975854 Mon Sep 17 00:00:00 2001 From: Bram Date: Sun, 28 May 2023 08:19:07 +0200 Subject: [PATCH] fix import --- tests/sensors/test_power.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/sensors/test_power.py b/tests/sensors/test_power.py index 3b195d2da..d39e85369 100644 --- a/tests/sensors/test_power.py +++ b/tests/sensors/test_power.py @@ -3,7 +3,7 @@ from unittest.mock import patch import pytest -from homeassistant.components.sensor import SensorStateClass +from homeassistant.components.sensor import ATTR_STATE_CLASS, SensorStateClass from homeassistant.components.utility_meter.sensor import SensorDeviceClass from homeassistant.components.vacuum import ( ATTR_BATTERY_LEVEL, @@ -29,7 +29,6 @@ from homeassistant.core import EVENT_HOMEASSISTANT_START, CoreState, HomeAssistant from homeassistant.setup import async_setup_component from homeassistant.util import dt -from plugwise.constants import ATTR_STATE_CLASS from pytest_homeassistant_custom_component.common import ( MockEntity, MockEntityPlatform,