VerifyGrahamCluster

Complexity

Total Complexity

4

Size/Duplication

Total Lines	21
Duplicated Lines	100 %

Importance

Changes	2
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	get_arguments()	15	15	4

import sys

from verify_slurm_cluster import VerifySlurmCluster, set_defaults


class VerifyGrahamCluster(VerifySlurmCluster):

This code seems to be duplicated in your project.

    WALLTIME = 60
    CORES_PER_NODE = 32
    GPUS_PER_NODE = 2

    def get_arguments(self, **kwargs):

        kwargs = super(VerifyGrahamCluster, self).get_arguments(**kwargs)

        if kwargs["gpusPerCommand"] == 0:
            account = os.environ.get("CPU_SLURM_ACCOUNT")
        else:
            account = os.environ.get("GPU_SLURM_ACCOUNT")

        if "sbatchFlags" not in kwargs or len(kwargs["sbatchFlags"]) == 0:
            kwargs["sbatchFlags"] = "--account=" + account
        else:
            kwargs["sbatchFlags"] += " --account=" + account

        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)