more flags which only work for function blocks

diff --git a/Python/compile.c b/Python/compile.c
index 917f9d3b12e564e58ff4173b6fa583b8fde7565c..a252bd15f10a7d4c694d2491e7bbd2f1c7295383 100644 (file)
--- a/Python/compile.c
+++ b/Python/compile.c
@@ -3794,13 +3794,11 @@ compute_code_flags(struct compiler *c)
                        flags |= CO_NESTED;
                if (ste->ste_generator)
                        flags |= CO_GENERATOR;
+               if (ste->ste_varargs)
+                       flags |= CO_VARARGS;
+               if (ste->ste_varkeywords)
+                       flags |= CO_VARKEYWORDS;
        }
-       if (ste->ste_varargs)
-               flags |= CO_VARARGS;
-       if (ste->ste_varkeywords)
-               flags |= CO_VARKEYWORDS;
-       if (ste->ste_generator)
-               flags |= CO_GENERATOR;
 
        /* (Only) inherit compilerflags in PyCF_MASK */
        flags |= (c->c_flags->cf_flags & PyCF_MASK);