dodox/Slate
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-06-24 20:42:02 +02:00
parent 1830c36fd8
commit ff1ba0be03
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
slate/slate.py
View File

@ -225,11 +225,13 @@ class Slate():
sh_lines += self.consume(slurmC, 'sh_lines', []) sh_lines += self.consume(slurmC, 'sh_lines', [])
if venv := self.consume(slurmC, 'venv', False): if venv := self.consume(slurmC, 'venv', False):
sh_lines += [f'source activate {venv}'] 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): if self.consume(slurmC, 'xvfb', False):
final_line = f'xvfb-run {final_line}' 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] sh_lines += [final_line]
script = "\n".join(sh_lines) script = "\n".join(sh_lines)