From 9095d8ec724087b6fe33242ae41a6e9d58566833 Mon Sep 17 00:00:00 2001
From: Fred McDavid <fred.mcdavid@zerocater.com>
Date: Thu, 19 Oct 2017 13:10:35 -0400
Subject: [PATCH] updates to process and pipe import code...need a way to turn
 billiard on/off

---
 memory_profiler.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/memory_profiler.py b/memory_profiler.py
index e0877f6..43c2f5e 100644
--- 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