diff --git a/damnit/backend/extract_data.py b/damnit/backend/extract_data.py index 14273544..f74ea305 100644 --- a/damnit/backend/extract_data.py +++ b/damnit/backend/extract_data.py @@ -152,10 +152,15 @@ def proposal(self): return self._proposal def slurm_options(self): + opts = ["--time", self.db.metameta.get("slurm_time", "02:00:00")] + if reservation := self.db.metameta.get('slurm_reservation', ''): - return ['--reservation', reservation] - partition = self.db.metameta.get('slurm_partition', '') or default_slurm_partition() - return ['--partition', partition] + opts.extend(['--reservation', reservation]) + else: + partition = self.db.metameta.get('slurm_partition', '') or default_slurm_partition() + opts.extend(['--partition', partition]) + + return opts def extract_and_ingest(self, proposal, run, cluster=False, run_data=RunData.ALL, match=()):