in submitit/core/core.py [0:0]
def task(self, task_id: int) -> "Job[R]":
"""Returns a given sub-Job (task).
Parameters
----------
task_id
The id of the task. Must be between 0 and self.num_tasks
Returns
-------
job
The sub_job. You can call all Job methods on it (done, stdout, ...)
If the job doesn't have sub jobs, return the job itself.
"""
if not 0 <= task_id < self.num_tasks:
raise ValueError(f"task_id {task_id} must be between 0 and {self.num_tasks - 1}")
if not self._sub_jobs:
return self
return self._sub_jobs[task_id]