Mercurial Repositories > eric / file diff

diff -r 9453efa25fd5 -r e4e2efb4846a Plugins/CheckerPlugins/Pep8/Pep257Checker.py
--- a/Plugins/CheckerPlugins/Pep8/Pep257Checker.py	Sun Sep 29 14:12:38 2013 +0200
+++ b/Plugins/CheckerPlugins/Pep8/Pep257Checker.py	Sun Sep 29 14:22:08 2013 +0200
@@ -1132,7 +1132,7 @@
         tokens = list(
             tokenize.generate_tokens(StringIO(context.ssource()).readline))
         return_ = [tokens[i + 1][0] for i,  token in enumerate(tokens)
-                   if token[1] == "return"]
+                   if token[1] in ("return", "yield")]
         if "@return" not in docstringContext.ssource():
             if (set(return_) -
                     set([tokenize.COMMENT, tokenize.NL, tokenize.NEWLINE]) !=