Mercurial Repositories > eric / file diff

diff -r 9453efa25fd5 -r e4e2efb4846a UtilitiesPython2/Pep257CheckerPy2.py
--- a/UtilitiesPython2/Pep257CheckerPy2.py	Sun Sep 29 14:12:38 2013 +0200
+++ b/UtilitiesPython2/Pep257CheckerPy2.py	Sun Sep 29 14:22:08 2013 +0200
@@ -1000,7 +1000,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]) !=