diff --git a/oargen.py b/oargen.py index b0460d41f7ab8dc916b5c447a8a21fedf008e015..5d501f995312f0da3646777ed432d1a27602d71b 100755 --- a/oargen.py +++ b/oargen.py @@ -19,8 +19,8 @@ def argparser(): help="Name to give to the job") parser.add_argument('-b', '--besteffort', action="store_true", help="Launch job in besteffort mode") - parser.add_argument('-t', '--time', default=[12], type=int, nargs="+", - help="Estimated maximum duration of the job (format: h [m [s]])") + parser.add_argument('-t', '--time', default="10", + help="Estimated maximum duration of the job (format: h[:m[:s]])") parser.add_argument('-g', '--gpu', action="store_true", help="If True, reserves only cores with GPUs") parser.add_argument('-c', '--core', default=1, type=int, @@ -62,9 +62,10 @@ def main(): command.append(properties) command.append("-l") - hour = args.time[0] - minutes = args.time[1] if len(args.time) >= 2 else 0 - seconds = args.time[2] if len(args.time) >= 3 else 0 + time = args.time.split(':') + hour = time[0] + minutes = time[1] if len(time) >= 2 else "00" + seconds = time[2] if len(time) >= 3 else "00" ressources = "core={},walltime={}:{}:{}".format(args.core, hour, minutes, seconds) command.append(ressources)