Skip to content

Commit 431a977

Browse files
committed
Remove redundant checks in AggregationHandler
1 parent d62e8d5 commit 431a977

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
@@ -406,10 +406,7 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
406406

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

0 commit comments

Comments
 (0)