Skip to content

Commit e97ac6a

Browse files
pre-commit-ci[bot]unkcpz
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent faa0ca0 commit e97ac6a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

aiida_hyperqueue/scheduler.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
class HyperQueueJobResource(JobResource):
3030
"""Class for HyperQueue job resources."""
3131

32-
_default_fields = ("num_cpus", "memory_mb")
32+
_default_fields = ("num_cpus", "memory_mb", "num_machines", "num_mpiprocs_per_machine")
3333

3434
_features = {
3535
"can_query_by_user": False,
@@ -58,9 +58,13 @@ def validate_resources(cls, **kwargs):
5858
resources.num_cpus = kwargs.pop("num_cpus")
5959
except KeyError:
6060
try:
61-
resources.num_cpus = kwargs.pop("num_machines") * kwargs.pop("num_mpiprocs_per_machine")
61+
resources.num_cpus = kwargs.pop("num_machines") * kwargs.pop(
62+
"num_mpiprocs_per_machine"
63+
)
6264
except KeyError:
63-
raise KeyError(f"Must specify `num_cpus`, or (`num_machines` and `num_mpiprocs_per_machine`) {kwargs}")
65+
raise KeyError(
66+
f"Must specify `num_cpus`, or (`num_machines` and `num_mpiprocs_per_machine`) {kwargs}"
67+
)
6468
else:
6569
if not isinstance(resources.num_cpus, int):
6670
raise ValueError("`num_cpus` must be an integer")

0 commit comments

Comments
 (0)