VerifyGrahamCluster

Complexity

Total Complexity

1

Size/Duplication

Total Lines	18
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	get_arguments()	0	12	1

import sys

from verify_slurm_cluster import VerifySlurmCluster, set_defaults


class VerifyGrahamCluster(VerifySlurmCluster):

    WALLTIME = 60
    CORES_PER_NODE = 32
    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,
            sbatchFlags="--account=rpp-bengioy")

        return kwargs


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