diff options
-rw-r--r-- | Cython/Compiler/Main.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/Cython/Compiler/Main.py b/Cython/Compiler/Main.py index 9465edab3..9130fbf19 100644 --- a/Cython/Compiler/Main.py +++ b/Cython/Compiler/Main.py @@ -98,16 +98,16 @@ class Context(object): self.gdb_debug_outputwriter = None def set_language_level(self, level): - from .Future import print_function, unicode_literals, absolute_import, division + from .Future import print_function, unicode_literals, absolute_import, division, generator_stop future_directives = [] if level == '3str': - future_directives = [print_function, absolute_import, division] + future_directives = [print_function, absolute_import, division, generator_stop] self.future_directives.discard(unicode_literals) level = 3 else: level = int(level) if level >= 3: - future_directives = [print_function, unicode_literals, absolute_import, division] + future_directives = [print_function, unicode_literals, absolute_import, division, generator_stop] self.language_level = level if future_directives: self.future_directives.update(future_directives) |