Enforce NumberDeviceClass in number entities (#83402)

2022-12-06 14:58:22 +01:00 · 2022-12-06 14:58:22 +01:00 · ba0d6fe611
commit ba0d6fe611
parent 647469f472
3 changed files with 14 additions and 3 deletions
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@ -1987,7 +1987,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
            matches=[
                TypeHintMatch(
                    function_name="device_class",
-                    return_type=["NumberDeviceClass", "str", None],
+                    return_type=["NumberDeviceClass", None],
                ),
                TypeHintMatch(
                    function_name="capability_attributes",