From a3765b2977790c96a39c573dd67de2187c38243c Mon Sep 17 00:00:00 2001 From: Robert Hillis Date: Thu, 16 Dec 2021 06:52:29 -0500 Subject: [PATCH] Use enums in updated (#62030) --- homeassistant/components/updater/binary_sensor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/updater/binary_sensor.py b/homeassistant/components/updater/binary_sensor.py index 10090946f6e..5ed7a7969ee 100644 --- a/homeassistant/components/updater/binary_sensor.py +++ b/homeassistant/components/updater/binary_sensor.py @@ -2,7 +2,7 @@ from __future__ import annotations from homeassistant.components.binary_sensor import ( - DEVICE_CLASS_UPDATE, + BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -21,7 +21,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info= class UpdaterBinary(CoordinatorEntity, BinarySensorEntity): """Representation of an updater binary sensor.""" - _attr_device_class = DEVICE_CLASS_UPDATE + _attr_device_class = BinarySensorDeviceClass.UPDATE _attr_name = "Updater" _attr_unique_id = "updater"