Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 18 additions & 18 deletions pympipool/interfaces/fluxbroker.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
get_future_done,
execute_task_dict,
)
from pympipool.interfaces.base import ExecutorBase
from pympipool.shared.base import ExecutorBase
from pympipool.shared.thread import RaisingThread
from pympipool.shared.taskexecutor import (
cloudpickle_register,
Expand All @@ -34,23 +34,23 @@ class SingleTaskExecutor(ExecutorBase):

Simple example:
```
import numpy as np
from pympipool import Executor

def calc(i, j, k):
from mpi4py import MPI
size = MPI.COMM_WORLD.Get_size()
rank = MPI.COMM_WORLD.Get_rank()
return np.array([i, j, k]), size, rank

def init_k():
return {"k": 3}

with Executor(cores=2, init_function=init_k) as p:
fs = p.submit(calc, 2, j=4)
print(fs.result())

>>> [(array([2, 4, 3]), 2, 0), (array([2, 4, 3]), 2, 1)]
>>> import numpy as np
>>> from pympipool import Executor
>>>
>>> def calc(i, j, k):
>>> from mpi4py import MPI
>>> size = MPI.COMM_WORLD.Get_size()
>>> rank = MPI.COMM_WORLD.Get_rank()
>>> return np.array([i, j, k]), size, rank
>>>
>>> def init_k():
>>> return {"k": 3}
>>>
>>> with Executor(cores=2, init_function=init_k) as p:
>>> fs = p.submit(calc, 2, j=4)
>>> print(fs.result())

[(array([2, 4, 3]), 2, 0), (array([2, 4, 3]), 2, 1)]
```
"""

Expand Down
2 changes: 1 addition & 1 deletion pympipool/interfaces/taskbroker.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from pympipool.interfaces.base import ExecutorBase
from pympipool.shared.base import ExecutorBase
from pympipool.shared.thread import RaisingThread
from pympipool.shared.broker import executor_broker

Expand Down
2 changes: 1 addition & 1 deletion pympipool/interfaces/taskexecutor.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from pympipool.interfaces.base import ExecutorBase
from pympipool.shared.base import ExecutorBase
from pympipool.shared.thread import RaisingThread
from pympipool.shared.taskexecutor import (
execute_parallel_tasks,
Expand Down
2 changes: 1 addition & 1 deletion pympipool/legacy/interfaces/executor.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from pympipool.interfaces.base import ExecutorBase
from pympipool.shared.base import ExecutorBase
from pympipool.shared.thread import RaisingThread
from pympipool.legacy.shared.interface import execute_serial_tasks
from pympipool.shared.taskexecutor import cloudpickle_register
Expand Down
File renamed without changes.