Merge pull request #272 from asmeurer/return-module-fix

Don't return early when "returning" from a module
inducer · Sep 3, 2017 · 676ee9d · 676ee9d
2 parents 6be804e + 6e285b6
commit 676ee9d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pudb/debugger.py b/pudb/debugger.py
@@ -382,10 +382,8 @@ def user_line(self, frame):
 
     def user_return(self, frame, return_value):
         """This function is called when a return trap is set here."""
-        if frame.f_code.co_name == '<module>':
-            return
-
-        frame.f_locals['__return__'] = return_value
+        if frame.f_code.co_name != '<module>':
+            frame.f_locals['__return__'] = return_value
 
         if self._wait_for_mainpyfile:
             if (self.mainpyfile != self.canonic(frame.f_code.co_filename)