Skip to content

Commit 2d157d8

Browse files
committed
Remove redundant checks in AggregationHandler
1 parent 02e2024 commit 2d157d8

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

pylint/pyreverse/inspector.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -407,10 +407,7 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
407407

408408
# If the element is a Name, it means it's an existing object, so it's aggregation
409409
if isinstance(element, nodes.Name):
410-
if isinstance(value, nodes.DictComp):
411-
element_type = safe_infer(value.value)
412-
else:
413-
element_type = safe_infer(value.elt)
410+
element_type = safe_infer(element)
414411
if element_type:
415412
current = set(parent.aggregations_type[node.attrname])
416413
parent.aggregations_type[node.attrname] = list(

0 commit comments

Comments
 (0)