Fix inheritance diagram from module

This commit is contained in:
Brent Yi 2020-09-07 23:09:19 -07:00 committed by Ashley Whetter
parent 9a5f22d7d6
commit f986f02105

View File

@ -52,7 +52,7 @@ def _import_class(name, currmodule):
if isinstance(target, astroid.Module): if isinstance(target, astroid.Module):
classes = [] classes = []
for child in target.children: for child in target.get_children():
if isinstance(child, astroid.ClassDef): if isinstance(child, astroid.ClassDef):
classes.append(child) classes.append(child)
return classes return classes