improved error-message when using @ex.automain in interactive mode

IDSIA · Apr 1, 2016 · 46b8f96 · 46b8f96
1 parent 04b6dfd
commit 46b8f96
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sacred/experiment.py b/sacred/experiment.py
@@ -85,6 +85,15 @@ def my_main():
         """
         captured = self.main(function)
         if function.__module__ == '__main__':
+            # Ensure that automain is not used in interactive mode.
+            import inspect
+            main_filename = inspect.getfile(function)
+            if (main_filename == '<stdin>' or
+                    (main_filename.startswith('<ipython-input-') and
+                     main_filename.endswith('>'))):
+                raise RuntimeError('Cannot use @ex.automain decorator in '
+                                   'interactive mode. Use @ex.main instead.')
+
             self.run_commandline()
         return captured