6786741185

Committed 07 Nov 2023 03:33PM UTC coverage: 97.076% (-0.3%) from 97.41%

Build # 6786741185

Build Type

Pull #207

github

Committed by

web-flow

Commit Message

Merge 64e8c392d into 74d8cea8e

Pull Request Pull Request #207: Universal executor

Run Details

9 of 11 new or added lines in 1 file covered. (81.82%)

1 existing line in 1 file now uncovered.

498 of 513 relevant lines covered (97.08%)

0.97 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

82.61

/pympipool/__init__.py

import os
from ._version import get_versions
from pympipool.mpi.executor import PyMPIExecutor

try:  # The PyFluxExecutor requires flux-core to be installed.
    from pympipool.flux.executor import PyFluxExecutor

    flux_installed = "FLUX_URI" in os.environ
except ImportError:
    flux_installed = False
    pass

try:  # The PySlurmExecutor requires the srun command to be available.
    from pympipool.slurm.executor import PySlurmExecutor

    slurm_installed = True
except ImportError:
    slurm_installed = False
    pass


__version__ = get_versions()["version"]
del get_versions


class Executor:
    def __new__(
        cls,
        max_workers,
        cores_per_worker=1,
        init_function=None,
        cwd=None,
        sleep_interval=0.1,
    ):
        if flux_installed:
            return PyFluxExecutor(
                max_workers=max_workers,
                cores_per_worker=cores_per_worker,
                init_function=init_function,
                cwd=cwd,
                sleep_interval=sleep_interval,
            )
        elif slurm_installed:
            return PySlurmExecutor(
                max_workers=max_workers,
                cores_per_worker=cores_per_worker,
                init_function=init_function,
                cwd=cwd,
                sleep_interval=sleep_interval,
            )
        else:
            return PyMPIExecutor(
                max_workers=max_workers,
                cores_per_worker=cores_per_worker,
                init_function=init_function,
                cwd=cwd,
                sleep_interval=sleep_interval,
            )

1	import os	1✔
2	from ._version import get_versions	1✔
3	from pympipool.mpi.executor import PyMPIExecutor	1✔
4
5	try: # The PyFluxExecutor requires flux-core to be installed.	1✔
6	from pympipool.flux.executor import PyFluxExecutor	1✔
7
8	flux_installed = "FLUX_URI" in os.environ	1✔
9	except ImportError:	×
NEW 10	flux_installed = False	×
UNCOV 11	pass	×
12
13	try: # The PySlurmExecutor requires the srun command to be available.	1✔
14	from pympipool.slurm.executor import PySlurmExecutor	1✔
15
NEW 16	slurm_installed = True	×
17	except ImportError:	1✔
18	slurm_installed = False	1✔
19	pass	1✔
20
21
22	__version__ = get_versions()["version"]	1✔
23	del get_versions	1✔
24
25
26	class Executor:	1✔
27	def __new__(	1✔
28	cls,
29	max_workers,
30	cores_per_worker=1,	1✔
31	init_function=None,	1✔
32	cwd=None,	1✔
33	sleep_interval=0.1,	1✔
34	):
35	if flux_installed:
36	return PyFluxExecutor(
37	max_workers=max_workers,
38	cores_per_worker=cores_per_worker,
39	init_function=init_function,
40	cwd=cwd,
41	sleep_interval=sleep_interval,
42	)
43	elif slurm_installed:
44	return PySlurmExecutor(
45	max_workers=max_workers,
46	cores_per_worker=cores_per_worker,
47	init_function=init_function,
48	cwd=cwd,
49	sleep_interval=sleep_interval,
50	)
51	else:
52	return PyMPIExecutor(
53	max_workers=max_workers,
54	cores_per_worker=cores_per_worker,
55	init_function=init_function,
56	cwd=cwd,
57	sleep_interval=sleep_interval,
58	)

pyiron / pympipool / 6786741185

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous