Completed
Pull Request — master (#173)
by
unknown
27s
created

VerifyMILACluster.get_arguments()   A

Complexity

Conditions 1

Size

Total Lines 11

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 1
c 1
b 0
f 0
dl 0
loc 11
rs 9.4285
1
import sys
2
3
from verify_slurm_cluster import VerifySlurmCluster, set_defaults
4
5
6
class VerifyMILACluster(VerifySlurmCluster):
7
8
    WALLTIME = 60
9
    CORES_PER_NODE = 8
10
    GPUS_PER_NODE = 2
11
12
    def get_arguments(self, **kwargs):
13
14
        set_defaults(
15
            kwargs,
16
            coresPerCommand=1,
17
            gpusPerCommand=0,
18
            walltime=self.WALLTIME,
19
            coresPerNode=self.CORES_PER_NODE,
20
            gpusPerNode=self.GPUS_PER_NODE)
21
22
        return kwargs
23
24
    def verify_simple_task_with_constraints(self, **kwargs):
25
26
        set_defaults(
27
            kwargs,
28
            gpusPerCommand=1,
29
            sbatchFlags='"-C\"gpu12gb\""')
30
31
        self.base_verification(**kwargs)
32
33
34
if __name__ == "__main__":
35
    verifications = filter(lambda o: not o.startswith("--"), sys.argv[1:])
36
    VerifyMILACluster(debug="--debug" in sys.argv[1:],
37
                      no_fork="--no-fork" in sys.argv[1:]).run_verifications(
38
        filtered_by=verifications)
39