updates to process and pipe import code...need a way to turn

billiard on/off
pythonprofilers · Oct 19, 2017 · 9095d8e · 9095d8e
1 parent 0f2658d
commit 9095d8e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/memory_profiler.py b/memory_profiler.py
@@ -36,9 +36,10 @@
         continue
     if use_if_test is None or use_if_test():
         try:
-            from multi_module import Process, Pipe
+            Process = multi_module.Process
+            Pipe = multi_module.Pipe
             break
-        except ImportError:
+        except AttributeError:
             continue