Allow float for int argument type [pylint plugin] (#114105)

2024-03-27 11:51:27 +01:00 · 2024-03-27 11:51:27 +01:00 · 3929273b41
commit 3929273b41
parent 5c97049f2e
4 changed files with 47 additions and 5 deletions
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@ -2968,6 +2968,15 @@ def _is_valid_type(
    ):
        return True

+    # Special case for int in argument type
+    if (
+        expected_type == "int"
+        and not in_return
+        and isinstance(node, nodes.Name)
+        and node.name in ("float", "int")
+    ):
+        return True
+
    # Name occurs when a namespace is not used, eg. "HomeAssistant"
    if isinstance(node, nodes.Name) and node.name == expected_type:
        return True