Replace some for statements with yield from

3 files changed, 4 insertions, 11 deletions

diff --git a/astroid/arguments.py b/astroid/arguments.py
index 508f4f85..d0eb40e8 100644
--- a/astroid/arguments.py
+++ b/astroid/arguments.py
@@ -12,7 +12,6 @@ from astroid import nodes
 from astroid import util
 
 
-
 class CallSite(object):
     """Class for understanding arguments passed into a call site
 
diff --git a/astroid/bases.py b/astroid/bases.py
index 9cf3eecd..b7778056 100644
--- a/astroid/bases.py
+++ b/astroid/bases.py
@@ -169,9 +169,7 @@ class BaseInstance(Proxy):
 
             # XXX frame should be self._proxied, or not ?
             get_attr = self.getattr(name, context, lookupclass=False)
-            for stmt in _infer_stmts(self._wrap_attr(get_attr, context),
-                                     context, frame=self):
-                yield stmt
+            yield from _infer_stmts(self._wrap_attr(get_attr, context), context, frame=self)
         except exceptions.AttributeInferenceError as error:
             try:
                 # fallback to class.igetattr since it has some logic to handle
@@ -180,8 +178,7 @@ class BaseInstance(Proxy):
                 if self._proxied.__class__.__name__ != 'ClassDef':
                     raise exceptions.InferenceError(**vars(error)) from error
                 attrs = self._proxied.igetattr(name, context, class_context=False)
-                for stmt in self._wrap_attr(attrs, context):
-                    yield stmt
+                yield from self._wrap_attr(attrs, context)
             except exceptions.AttributeInferenceError as error:
                 raise exceptions.InferenceError(**vars(error)) from error
 
diff --git a/astroid/builder.py b/astroid/builder.py
index 968aa410..979d3fcc 100644
--- a/astroid/builder.py
+++ b/astroid/builder.py
@@ -17,7 +17,6 @@ import os
 import sys
 import textwrap
 
-
 from astroid._ast import _parse
 from astroid import bases
 from astroid import exceptions
@@ -313,8 +312,7 @@ def _extract_expressions(node):
         yield real_expr
     else:
         for child in node.get_children():
-            for result in _extract_expressions(child):
-                yield result
+            yield from _extract_expressions(child)
 
 
 def _find_statement_by_line(node, line):
@@ -420,8 +418,7 @@ def extract_node(code, module_name=''):
     tree = parse(code, module_name=module_name)
     extracted = []
     if requested_lines:
-        for line in requested_lines:
-            extracted.append(_find_statement_by_line(tree, line))
+        extracted = [_find_statement_by_line(tree, line) for line in requested_lines]
 
     # Modifies the tree.
     extracted.extend(_extract_expressions(tree))