summaryrefslogtreecommitdiff
path: root/astroid/brain/brain_dataclasses.py
diff options
context:
space:
mode:
authorDaniƫl van Noord <13665637+DanielNoord@users.noreply.github.com>2022-02-13 01:50:27 +0100
committerGitHub <noreply@github.com>2022-02-13 01:50:27 +0100
commitcfd9e74f7b4cbac08357cadec03c736501368afa (patch)
tree290f364ff1861f43b7620260a36f7e156090be8d /astroid/brain/brain_dataclasses.py
parent552f1c19aab0719e7b30bb02e032d4bfe655f343 (diff)
downloadastroid-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.py1
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