Skip to content

Commit

Permalink
parallel magic: set nodes in cluster their cluster_size and cluster_r…
Browse files Browse the repository at this point in the history
…ank; as result, don't show list of all Nones (returns None)
  • Loading branch information
Doug Blank committed Sep 12, 2014
1 parent eeeb1c9 commit 486dba5
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions jupyter_kernel/magics/parallel_magic.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,12 @@ def line_parallel(self, module_name, class_name, kernel_name="default", ids=None
"class_name": class_name,
"kernel_name": kernel_name},
block=True)

self.view["kernels['%s'].set_variable(\"cluster_size\", %s)" % (
kernel_name, len(self.client))]
self.client[:].scatter('cluster_rank', self.client.ids, flatten=True)
self.view["kernels['%s'].set_variable(\"cluster_rank\", cluster_rank)" % (
kernel_name)]
self.retval = None

@option(
Expand Down Expand Up @@ -133,6 +139,8 @@ def cell_px(self, kernel_name=None, evaluate=False):
self.evaluate = evaluate

def post_process(self, retval):
if isinstance(self.retval, list) and not any(self.retval):
return None
return self.retval

def register_magics(kernel):
Expand Down

0 comments on commit 486dba5

Please sign in to comment.