Skip to content

Commit 73f956d

Browse files
adds code to enable profiling of Celery tasks
* as a workaround for multiprocessing refusal to fork daemon processes, attempts to import Process and Pipe from billiard, the Celery project's fork of the multiprocessing module
1 parent e2a97b9 commit 73f956d

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

memory_profiler.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,13 @@
2222

2323
# TODO: provide alternative when multiprocessing is not available
2424
try:
25-
from multiprocessing import Process, Pipe
25+
from billiard import Process, Pipe
2626
except ImportError:
27-
from multiprocessing.dummy import Process, Pipe
27+
try:
28+
from multiprocessing import Process, Pipe
29+
except ImportError:
30+
from multiprocessing.dummy import Process, Pipe
31+
2832

2933
try:
3034
from IPython.core.magic import Magics, line_cell_magic, magics_class

0 commit comments

Comments
 (0)