diff --git a/src/xfel_calibrate/repeat.py b/src/xfel_calibrate/repeat.py index 18fba8c43b1aeec04e16ac7712a32f21725a9314..b386ec9724fa1b3f1c12d033cb481f8970ddfc69 100644 --- a/src/xfel_calibrate/repeat.py +++ b/src/xfel_calibrate/repeat.py @@ -66,6 +66,7 @@ def main(argv=None): ap.add_argument("--out-folder", help="Directory to put output data") ap.add_argument("--report-path", help="Location to save PDF report") ap.add_argument("--slurm-partition", help="Submit jobs in this Slurm partition") + ap.add_argument("--slurm-mem", type=int, help="Requested node RAM in GB") ap.add_argument('--no-cluster-job', action="store_true", help="Run notebooks here, not in cluster jobs") args = ap.parse_args(argv) @@ -114,6 +115,7 @@ def main(argv=None): else: joblist = job_chain.submit_jobs(SlurmOptions( partition=args.slurm_partition, + mem=args.slurm_mem, )) fmt_args = {'run_path': working_dir,