Moving multiprocessing.freeze_support() to __main__.py. RE:natcap#1622

phargogh · Sep 11, 2024 · 1bbbfb2 · 1bbbfb2
1 parent 948806c
commit 1bbbfb2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/exe/hooks/rthook.py b/exe/hooks/rthook.py
@@ -1,9 +1,6 @@
-import sys
 import os
-import multiprocessing
 import platform
-
-multiprocessing.freeze_support()
+import sys
 
 os.environ['PROJ_LIB'] = os.path.join(sys._MEIPASS, 'proj')
 

diff --git a/src/natcap/invest/__main__.py b/src/natcap/invest/__main__.py
@@ -1,5 +1,11 @@
+import multiprocessing
 import sys
 
+# We want to guarantee that this is called BEFORE any other processes start,
+# which could happen at import time.
+if __name__ == '__main__':
+    multiprocessing.freeze_support()
+
 from . import cli
 
 if __name__ == '__main__':