Mark button methods and properties as mandatory in pylint plugin (#145269)

2025-05-20 11:48:33 +02:00
parent 642dc5b49c
commit a8264ae8ae
2 changed files with 3 additions and 2 deletions
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@@ -941,12 +941,13 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
            matches=[
                TypeHintMatch(
                    function_name="device_class",
-                    return_type=["ButtonDeviceClass", "str", None],
+                    return_type=["ButtonDeviceClass", None],
                ),
                TypeHintMatch(
                    function_name="press",
                    return_type=None,
                    has_async_counterpart=True,
+                    mandatory=True,
                ),
            ],
        ),