Skip to content

Commit

Permalink
Merge pull request #171 from pyiron/docstring
Browse files Browse the repository at this point in the history
Add Docstrings
  • Loading branch information
jan-janssen authored Aug 11, 2023
2 parents fa04783 + edede69 commit 4271d05
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 0 deletions.
12 changes: 12 additions & 0 deletions pympipool/flux/fluxbroker.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,18 @@


class PyFluxExecutor(ExecutorBase):
"""
Args:
max_workers (int): defines the number workers which can execute functions in parallel
cores_per_worker (int): number of MPI cores to be used for each function call
threads_per_core (int): number of OpenMP threads to be used for each function call
gpus_per_worker (int): number of GPUs per worker - defaults to 0
init_function (None): optional function to preset arguments for functions which are submitted later
cwd (str/None): current working directory where the parallel python task is executed
sleep_interval (float): synchronization interval - default 0.1
executor (flux.job.FluxExecutor): Flux Python interface to submit the workers to flux
"""

def __init__(
self,
max_workers,
Expand Down
13 changes: 13 additions & 0 deletions pympipool/mpi/mpibroker.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,19 @@


class PyMPIExecutor(ExecutorBase):
"""
Args:
max_workers (int): defines the number workers which can execute functions in parallel
cores_per_worker (int): number of MPI cores to be used for each function call
threads_per_core (int): number of OpenMP threads to be used for each function call
gpus_per_worker (int): number of GPUs per worker - defaults to 0
oversubscribe (bool): adds the `--oversubscribe` command line flag (OpenMPI only) - default False
init_function (None): optional function to preset arguments for functions which are submitted later
cwd (str/None): current working directory where the parallel python task is executed
sleep_interval (float): synchronization interval - default 0.1
enable_slurm_backend (bool): enable the SLURM queueing system as backend - defaults to False
"""

def __init__(
self,
max_workers,
Expand Down

0 comments on commit 4271d05

Please sign in to comment.