diff --git a/qsub-wrap.py b/qsub-wrap.py index 44c24d5553c38cd2fd1ac91b3c415c350c97b17e..4692347880472105e2644cf266f0b3cb1e73a702 100755 --- a/qsub-wrap.py +++ b/qsub-wrap.py @@ -152,7 +152,7 @@ def addvariableargs(parser): return kwargs_pass, parser -def submit_pbs(walltime='00:59:59', queue = 'xpresq', py_env='anaconda', +def submit_pbs(walltime='00:59:59', queue='xpresq', py_env='wetb_py3', pbs_in='pbs_in', pbs_out='pbs_out', jobname=None, command=None, fpath=None, dry=False, **kwargs): """ @@ -254,6 +254,15 @@ if __name__ == '__main__': default='00:29:59') parser.add_argument('--dry', action='store_true', dest='dry', default=False, help='dry run: do not submit pbs file.') + parser.add_argument('--jobname', type='string', action='store', + dest='jobname', default=None, + help='PBS job name, default: python file name.') + parser.add_argument('--pbs_in', type='string', action='store', + dest='pbs_in', default=None, + help='pbs_in directory, default: pbs_in') + parser.add_argument('--pbs_out', type='string', action='store', + dest='pbs_out', default=None, + help='pbs_out directory, default: pbs_out') # get all the other arguments and add them to the help kwargs_pass, parser = addvariableargs(parser) @@ -270,4 +279,5 @@ if __name__ == '__main__': pbs_script, stdout = submit_pbs(fpath=opt.fpath, command=opt.command, py_env=opt.py_env, queue=opt.queue, walltime=opt.walltime, dry=opt.dry, - **kwargs_pass) + jobname=opt.jobname, pbs_in=opt.pbs_in, + pbs_out=opt.pbs_out, **kwargs_pass)