Made TreeArguments methods a bit more understandable

This commit is contained in:
Dave Halter
2019-05-08 09:30:39 +02:00
parent dc15470e0b
commit 037a069ddd
2 changed files with 13 additions and 14 deletions

View File

@@ -301,13 +301,7 @@ class Script(object):
names = list(self._evaluator.goto(context, tree_name))
if follow_imports:
def check(name):
return name.is_import()
else:
def check(name):
return False
names = filter_follow_imports(names, check)
names = filter_follow_imports(names, lambda name: name.is_import())
names = try_stub_to_actual_names(names, prefer_stub_to_compiled=True)
defs = [classes.Definition(self._evaluator, d) for d in set(names)]