Make __mp_main__ an alias for __main__ in all processes to simplify

pickling of classes defined in main module.
This commit is contained in:
Richard Oudkerk 2012-10-08 13:07:00 +01:00
parent c3c6fe5b1a
commit 5046e9741c
2 changed files with 13 additions and 16 deletions

View file

@ -39,6 +39,13 @@ import sys
from multiprocessing.process import Process, current_process, active_children
from multiprocessing.util import SUBDEBUG, SUBWARNING
#
# Alias for main module -- will be reset by bootstrapping child processes
#
if '__main__' in sys.modules:
sys.modules['__mp_main__'] = sys.modules['__main__']
#
# Exceptions
#