submitit_basic_slurm_job_submission_with_autoexecutor.py

1import submitit
2
3def add(a, b):
4    return a + b
5
6# executor is the object that will manage the jobs
7# Here we use a local executor for testing, change it to SlurmExecutor to submit to a cluster
8executor = submitit.AutoExecutor(folder="log_test")
9
10# set the parameters for the executor
11executor.update_parameters(timeout_min=1, slurm_partition="dev")
12
13# submit the job
14job = executor.submit(add, 5, 7)
15
16# wait for the job to finish and get the result
17print(job.result())