Some minor cleanups in WLED (#117453)

This commit is contained in:
Franck Nijhof 2024-05-14 21:02:31 +02:00 committed by GitHub
parent 0df9006bf7
commit faff5f4738
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 13 additions and 16 deletions

View file

@ -9,8 +9,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import WLEDConfigEntry
from .coordinator import WLEDDataUpdateCoordinator
from .entity import WLEDEntity
from .helpers import wled_exception_handler
from .models import WLEDEntity
async def async_setup_entry(
@ -19,8 +19,7 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up WLED button based on a config entry."""
coordinator = entry.runtime_data
async_add_entities([WLEDRestartButton(coordinator)])
async_add_entities([WLEDRestartButton(entry.runtime_data)])
class WLEDRestartButton(WLEDEntity, ButtonEntity):