From 093202f138aa002f8330be63c1bbe8810329c9a4 Mon Sep 17 00:00:00 2001 From: Robert Hillis Date: Thu, 16 Dec 2021 08:41:59 -0500 Subject: [PATCH] Use enum in roku (#62043) --- homeassistant/components/roku/media_player.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/roku/media_player.py b/homeassistant/components/roku/media_player.py index bb9b4bfa37f..72b93994a10 100644 --- a/homeassistant/components/roku/media_player.py +++ b/homeassistant/components/roku/media_player.py @@ -7,9 +7,8 @@ import logging import voluptuous as vol from homeassistant.components.media_player import ( - DEVICE_CLASS_RECEIVER, - DEVICE_CLASS_TV, BrowseMedia, + MediaPlayerDeviceClass, MediaPlayerEntity, ) from homeassistant.components.media_player.const import ( @@ -104,9 +103,9 @@ class RokuMediaPlayer(RokuEntity, MediaPlayerEntity): def device_class(self) -> str | None: """Return the class of this device.""" if self.coordinator.data.info.device_type == "tv": - return DEVICE_CLASS_TV + return MediaPlayerDeviceClass.TV - return DEVICE_CLASS_RECEIVER + return MediaPlayerDeviceClass.RECEIVER @property def state(self) -> str | None: