From a1aca208189f0142adf2d81aff5e5a37d5d5e5b5 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Wed, 8 Sep 2021 22:48:31 +0200 Subject: [PATCH] Address review comment from #55833 (#55985) --- homeassistant/components/google_assistant/trait.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/homeassistant/components/google_assistant/trait.py b/homeassistant/components/google_assistant/trait.py index 393f8b22fbe..b2af1d6f9a9 100644 --- a/homeassistant/components/google_assistant/trait.py +++ b/homeassistant/components/google_assistant/trait.py @@ -2314,10 +2314,7 @@ class SensorStateTrait(_Trait): @classmethod def supported(cls, domain, features, device_class, _): """Test if state is supported.""" - return ( - domain == sensor.DOMAIN - and device_class in SensorStateTrait.sensor_types.keys() - ) + return domain == sensor.DOMAIN and device_class in cls.sensor_types def sync_attributes(self): """Return attributes for a sync request."""