Support for xvfb-run wrapper

This commit is contained in:
Dominik Moritz Roth 2024-02-02 18:16:46 +01:00
parent 03d2ab1733
commit 9762e50987

View File

@ -225,7 +225,10 @@ class Slate():
sh_lines += self.consume(slurmC, 'sh_lines', [])
if venv := self.consume(slurmC, 'venv', False):
sh_lines += [f'source activate {venv}']
sh_lines += [f'python3 {python_script} {filename} {name} -t $SLURM_ARRAY_TASK_ID --sweep_id {self.sweep_id}']
if self.consume(slurmC, 'xvfb', False):
sh_lines += [f'xvfb-run python3 {python_script} {filename} {name} -t $SLURM_ARRAY_TASK_ID --sweep_id {self.sweep_id}']
else:
sh_lines += [f'python3 {python_script} {filename} {name} -t $SLURM_ARRAY_TASK_ID --sweep_id {self.sweep_id}']
script = "\n".join(sh_lines)
num_jobs = self._calc_num_jobs(schedC, num_conv_versions)