diff --git a/slate/slate.py b/slate/slate.py index f4d0ea2..f478022 100644 --- a/slate/slate.py +++ b/slate/slate.py @@ -225,10 +225,12 @@ class Slate(): sh_lines += self.consume(slurmC, 'sh_lines', []) if venv := self.consume(slurmC, 'venv', False): sh_lines += [f'source activate {venv}'] + final_line = 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}'] + final_line = f'xvfb-run {final_line}' + if self.consume(slurmC, 'isaac_sif', False): + final_line = f'./omni_sif_python {final_line}' + sh_lines += [final_line] script = "\n".join(sh_lines) num_jobs = self._calc_num_jobs(schedC, num_conv_versions)