fix(switch): add device_class

alandtse · Jan 18, 2020 · 016c22e · 016c22e
1 parent cd27bb9
commit 016c22e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/custom_components/alexa_media/switch.py b/custom_components/alexa_media/switch.py
@@ -231,6 +231,11 @@ def name(self):
         """Return the name of the switch."""
         return "{} {} switch".format(self._client.name, self._name)
 
+    @property
+    def device_class(self):
+        """Return the device_class of the switch."""
+        return "switch"
+
     @property
     def should_poll(self):
         """Return the polling state."""