Enforce SirenEntityFeature (#82464)

home-assistant · Nov 22, 2022 · ff5f1ae · ff5f1ae
1 parent 0cb3ec1
commit ff5f1ae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/homeassistant/components/siren/__init__.py b/homeassistant/components/siren/__init__.py
@@ -162,7 +162,7 @@ class SirenEntity(ToggleEntity):
 
     entity_description: SirenEntityDescription
     _attr_available_tones: list[int | str] | dict[int, str] | None
-    _attr_supported_features: SirenEntityFeature | int = 0
+    _attr_supported_features: SirenEntityFeature = SirenEntityFeature(0)
 
     @final
     @property
@@ -190,6 +190,6 @@ def available_tones(self) -> list[int | str] | dict[int, str] | None:
         return None
 
     @property
-    def supported_features(self) -> SirenEntityFeature | int:
+    def supported_features(self) -> SirenEntityFeature:
         """Return the list of supported features."""
         return self._attr_supported_features
diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py
@@ -2214,7 +2214,7 @@ class ClassTypeHintMatch:
                 ),
                 TypeHintMatch(
                     function_name="supported_features",
-                    return_type=["SirenEntityFeature", "int"],
+                    return_type="SirenEntityFeature",
                 ),
             ],
         ),