Some minor cleanups in WLED (#117453)
This commit is contained in:
parent
0df9006bf7
commit
faff5f4738
10 changed files with 13 additions and 16 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue