Conditions | 4 |
Total Lines | 15 |
Lines | 15 |
Ratio | 100 % |
Changes | 2 | ||
Bugs | 0 | Features | 0 |
1 | import sys |
||
12 | def get_arguments(self, **kwargs): |
||
13 | |||
14 | kwargs = super(VerifyGrahamCluster, self).get_arguments(**kwargs) |
||
15 | |||
16 | if kwargs["gpusPerCommand"] == 0: |
||
17 | account = os.environ.get("CPU_SLURM_ACCOUNT") |
||
18 | else: |
||
19 | account = os.environ.get("GPU_SLURM_ACCOUNT") |
||
20 | |||
21 | if "sbatchFlags" not in kwargs or len(kwargs["sbatchFlags"]) == 0: |
||
22 | kwargs["sbatchFlags"] = "--account=" + account |
||
23 | else: |
||
24 | kwargs["sbatchFlags"] += " --account=" + account |
||
25 | |||
26 | return kwargs |
||
27 | |||
34 |