parallelfunction.py
31 lines
| 1.2 KiB
| text/x-python
|
PythonLexer
Brian E Granger
|
r1234 | # encoding: utf-8 | |
| """A parallelized function that does scatter/execute/gather.""" | |||
| __docformat__ = "restructuredtext en" | |||
| #------------------------------------------------------------------------------- | |||
| # Copyright (C) 2008 The IPython Development Team | |||
| # | |||
| # Distributed under the terms of the BSD License. The full license is in | |||
| # the file COPYING, distributed as part of this software. | |||
| #------------------------------------------------------------------------------- | |||
| #------------------------------------------------------------------------------- | |||
| # Imports | |||
| #------------------------------------------------------------------------------- | |||
| from types import FunctionType | |||
| class ParallelFunction: | |||
| """A function that operates in parallel on sequences.""" | |||
| def __init__(self, func, multiengine, targets, block): | |||
| """Create a `ParallelFunction`. | |||
| """ | |||
| assert isinstance(func, (str, FunctionType)), "func must be a fuction or str" | |||
| self.func = func | |||
| self.multiengine = multiengine | |||
| self.targets = targets | |||
| self.block = block | |||
| def __call__(self, sequence): | |||
| return self.multiengine.map(self.func, sequence, targets=self.targets, block=self.block) |