VerifyMILACluster

Complexity

Total Complexity

2

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	get_arguments()	0	11	1
A	verify_simple_task_with_constraints()	0	8	1

import sys

from verify_slurm_cluster import VerifySlurmCluster, set_defaults


class VerifyMILACluster(VerifySlurmCluster):

    WALLTIME = 60
    CORES_PER_NODE = 8
    GPUS_PER_NODE = 2

    def get_arguments(self, **kwargs):

        set_defaults(
            kwargs,
            coresPerCommand=1,
            gpusPerCommand=0,
            walltime=self.WALLTIME,
            coresPerNode=self.CORES_PER_NODE,
            gpusPerNode=self.GPUS_PER_NODE)

        return kwargs

    def verify_simple_task_with_constraints(self, **kwargs):

        set_defaults(
            kwargs,
            gpusPerCommand=1,
            sbatchFlags='"-C\"gpu12gb\""')

        self.base_verification(**kwargs)


if __name__ == "__main__":
    verifications = filter(lambda o: not o.startswith("--"), sys.argv[1:])
    VerifyMILACluster(debug="--debug" in sys.argv[1:],
                      no_fork="--no-fork" in sys.argv[1:]).run_verifications(
        filtered_by=verifications)