diff options
author | Daniƫl van Noord <13665637+DanielNoord@users.noreply.github.com> | 2022-02-13 01:50:27 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-02-13 01:50:27 +0100 |
commit | cfd9e74f7b4cbac08357cadec03c736501368afa (patch) | |
tree | 290f364ff1861f43b7620260a36f7e156090be8d /astroid/brain/brain_dataclasses.py | |
parent | 552f1c19aab0719e7b30bb02e032d4bfe655f343 (diff) | |
download | astroid-git-cfd9e74f7b4cbac08357cadec03c736501368afa.tar.gz |
Add is_dataclass attribute to ClassDef (#1391)
Diffstat (limited to 'astroid/brain/brain_dataclasses.py')
-rw-r--r-- | astroid/brain/brain_dataclasses.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/astroid/brain/brain_dataclasses.py b/astroid/brain/brain_dataclasses.py index bfdbbe09..a667e80d 100644 --- a/astroid/brain/brain_dataclasses.py +++ b/astroid/brain/brain_dataclasses.py @@ -67,6 +67,7 @@ def is_decorated_with_dataclass(node, decorator_names=DATACLASSES_DECORATORS): def dataclass_transform(node: ClassDef) -> None: """Rewrite a dataclass to be easily understood by pylint""" + node.is_dataclass = True for assign_node in _get_dataclass_attributes(node): name = assign_node.target.name |