diff --git a/homeassistant/components/blebox/button.py b/homeassistant/components/blebox/button.py index 01de5fa56b7..e9ceaac2dc7 100644 --- a/homeassistant/components/blebox/button.py +++ b/homeassistant/components/blebox/button.py @@ -1,7 +1,7 @@ """BleBox button entities implementation.""" from __future__ import annotations -from homeassistant.components.button import ButtonDeviceClass, ButtonEntity +from homeassistant.components.button import ButtonEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -26,7 +26,6 @@ class BleBoxButtonEntity(BleBoxEntity, ButtonEntity): def __init__(self, feature): """Initialize a BleBox button feature.""" super().__init__(feature) - self._attr_device_class = ButtonDeviceClass.UPDATE self._attr_icon = self.get_icon() def get_icon(self): diff --git a/tests/components/blebox/test_button.py b/tests/components/blebox/test_button.py index 22e1b8cb734..8d89f31d260 100644 --- a/tests/components/blebox/test_button.py +++ b/tests/components/blebox/test_button.py @@ -5,7 +5,6 @@ from unittest.mock import PropertyMock import blebox_uniapi import pytest -from homeassistant.components.button import ButtonDeviceClass from homeassistant.const import ATTR_ICON from .conftest import async_setup_entity, mock_feature @@ -50,8 +49,6 @@ async def test_tvliftbox_init(tvliftbox, hass, config, caplog): assert entry.unique_id == "BleBox-tvLiftBox-4a3fdaad90aa-open_or_stop" - assert state.attributes["device_class"] == ButtonDeviceClass.UPDATE - assert state.name == "tvLiftBox-open_or_stop"