diff options
Diffstat (limited to 'astroid')
-rw-r--r-- | astroid/brain/brain_attrs.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/astroid/brain/brain_attrs.py b/astroid/brain/brain_attrs.py index acb069e3..7afcc8ab 100644 --- a/astroid/brain/brain_attrs.py +++ b/astroid/brain/brain_attrs.py @@ -8,6 +8,7 @@ Astroid hook for the attrs library Without this hook pylint reports unsupported-assignment-operation for attrs classes """ +from astroid.helpers import safe_infer from astroid.manager import AstroidManager from astroid.nodes.node_classes import AnnAssign, Assign, AssignName, Call, Unknown from astroid.nodes.scoped_nodes import ClassDef @@ -40,6 +41,10 @@ def is_decorated_with_attrs(node, decorator_names=ATTRS_NAMES) -> bool: decorator_attribute = decorator_attribute.func if decorator_attribute.as_string() in decorator_names: return True + + inferred = safe_infer(decorator_attribute) + if inferred and inferred.root().name == "attr._next_gen": + return True return False |