Skip to content

Commit ac1e323

Browse files
updates to process and pipe import code...need a way to turn
billiard on/off
1 parent 34b3b3e commit ac1e323

1 file changed

Lines changed: 9 additions & 6 deletions

File tree

memory_profiler.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,18 +18,21 @@
1818
import logging
1919
import traceback
2020
from signal import SIGKILL
21-
21+
import importlib
2222

2323
# TODO: provide alternative when multiprocessing is not available
24-
import billiard, multiprocessing
2524
MULTILIBS = (
26-
(billiard, lambda: True if "MEMPROF_PREFERS_BILLIARD" in os.environ and \
25+
("billiard", lambda: True if "MEMPROF_PREFERS_BILLIARD" in os.environ and \
2726
os.environ["MEMPROF_PREFERS_BILLIARD"] in ["True", "true", "TRUE", "1"]),
28-
(multiprocessing,),
29-
(multiprocessing.dummy,)
27+
("multiprocessing",),
28+
("multiprocessing.dummy",)
3029
)
3130
for module_tuple in MULTILIBS:
32-
multi_module, use_if_test = module_tuple
31+
multi_module_name, use_if_test = module_tuple
32+
try:
33+
multi_module = importlib.import_module(multi_module_name)
34+
except ImportError:
35+
continue
3336
if use_if_test is None or use_if_test():
3437
try:
3538
from multi_module import Process, Pipe

0 commit comments

Comments
 (0)