Merge pull request #25 from exabiome/enh/add_project_arg

exabiome · Feb 11, 2021 · a7a5279 · a7a5279
2 parents a36d300 + 68593e1
commit a7a5279
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/exabiome/run/cori.py b/src/exabiome/run/cori.py
@@ -13,6 +13,7 @@ class SlurmJob(AbstractJob):
     submit_cmd = 'sbatch'
     job_var = 'SLURM_JOB_ID'
     job_fmt_var = 'j'
+    job_id_re = 'Submitted batch job (\d+)'
 
     debug_queue = 'debug'
 

diff --git a/src/exabiome/run/run_job.py b/src/exabiome/run/run_job.py
@@ -47,6 +47,7 @@ def run_train(argv=None):
     rsc_grp.add_argument('-g', '--gpus',       help="the number of GPUs to use", default=None, type=int)
     rsc_grp.add_argument('-N', '--jobname',    help="the name of the job", default=None)
     rsc_grp.add_argument('-q', '--queue',      help="the queue to submit to", default=None)
+    rsc_grp.add_argument('-P', '--project',    help="the project/account to submit under", default=None)
 
     system_grp = parser.add_argument_group('Compute system')
     grp = system_grp.add_mutually_exclusive_group()
@@ -96,6 +97,9 @@ def run_train(argv=None):
 
     if args.queue is not None:
         job.queue = args.queue
+
+    if args.project is not None:
+        job.project = args.project
 
     args.input = os.path.abspath(args.input)