From af76e5dd981542eb08f86a40f74d6cc1694bd6ff Mon Sep 17 00:00:00 2001
From: Thomas Kluyver <thomas@kluyver.me.uk>
Date: Wed, 20 Apr 2022 17:07:04 +0100
Subject: [PATCH] Add --slurm-mem option to xfel-calibrate-repeat
---
src/xfel_calibrate/repeat.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/xfel_calibrate/repeat.py b/src/xfel_calibrate/repeat.py
index 18fba8c43..b386ec972 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,
--
GitLab