From ff1ba0be0379da9cbe71052a39a66b5597763c48 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Mon, 24 Jun 2024 20:42:02 +0200 Subject: [PATCH] . --- slate/slate.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/slate/slate.py b/slate/slate.py index f478022..cea4530 100644 --- a/slate/slate.py +++ b/slate/slate.py @@ -225,11 +225,13 @@ 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}' + final_line = f'{python_script} {filename} {name} -t $SLURM_ARRAY_TASK_ID --sweep_id {self.sweep_id}' + if self.consume(slurmC, 'python_exec', False): + final_line = f'./omni_sif_python {final_line}' + else: + final_line = f'python3 {final_line}' if self.consume(slurmC, 'xvfb', False): 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)