parallel magics: remove hack; fixed

Calysto · Sep 12, 2014 · f80e956 · f80e956
1 parent 543a3cd
commit f80e956
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/jupyter_kernel/magics/parallel_magic.py b/jupyter_kernel/magics/parallel_magic.py
@@ -155,14 +155,11 @@ def line_pmap(self, function_name, args, kernel_name=None):
         if kernel_name is None:
             kernel_name = self.kernel_name
 
-        # FIXME: horrible hack
-        self.view.execute("""
-import os
-os.kernels = kernels
-""")
-        self.retval = self.view_load_balanced.map_async(
-            lambda arg, kname=kernel_name, fname=function_name: os.kernels[kname].do_function_direct(fname, arg),
-            eval(args))
+        # To make sure we can find `kernels`:
+        from IPython.parallel.util import interactive
+        f = interactive(lambda arg, kname=kernel_name, fname=function_name: \
+                        kernels[kname].do_function_direct(fname, arg))
+        self.retval = self.view_load_balanced.map_async(f, eval(args))
 
     def post_process(self, retval):
         try: