def filter_is_done(self, tasks):
"""
Yield a dictionary of Slurm job metadata for each task that has completed.
"""
# jobid can be none if submission fialed
job_ids = [t.drm_jobID for t in tasks if t.drm_jobID is not None]
if job_ids:
job_infos = retry_call(do_sacct,
fargs=[job_ids],
delay=10,
tries=10,
backoff=2,
max_delay=60,
logger=tasks[0].workflow.log)
for task in tasks:
if task.drm_jobID in job_infos:
job_info = job_infos[task.drm_jobID]
if job_info['State'] in FAILED_STATES + COMPLETED_STATES:
job_info = parse_sacct(job_infos[task.drm_jobID],
tasks[0].workflow.log)
yield task, job_info
else:
assert job_info[
'State'] in PENDING_STATES, 'Invalid job state: `%s` for %s drm_job_id=%s' % (
job_info['State'], task, task.drm_jobID)
def submit_job(self, task):
for p in [task.output_stdout_path, task.output_stderr_path]:
if os.path.exists(p):
os.unlink(p)
task.drm_jobID = retry_call(sbatch, fargs=[task],
delay=10, tries=10, backoff=2, max_delay=60,
logger=task.log)
task.status = TaskStatus.submitted
def submit_job(self, task):
if task.environment_variables is not None:
raise NotImplementedError
for p in [task.output_stdout_path, task.output_stderr_path]:
if os.path.exists(p):
os.unlink(p)
task.drm_jobID = retry_call(
sbatch, fargs=[task], delay=10, tries=10, backoff=2, max_delay=60, logger=task.log,
)
task.status = TaskStatus.submitted
def submit_job(self, task):
for p in [task.output_stdout_path, task.output_stderr_path]:
if os.path.exists(p):
os.unlink(p)
task.drm_jobID = retry_call(sbatch,
fargs=[task],
delay=10,
tries=10,
backoff=2,
max_delay=60,
logger=task.log)
task.status = TaskStatus.submitted
def drm_statuses(self, tasks, log_errors=True):
"""
:param tasks: tasks that have been submitted to the job manager
:returns: (dict) task.drm_jobID -> drm_status
"""
job_ids = [t.drm_jobID for t in tasks if t.drm_jobID is not None]
if job_ids:
job_infos = retry_call(do_sacct, fargs=[job_ids],
delay=10, tries=10, backoff=2, max_delay=60,
logger=tasks[0].workflow.log)
def f(task):
return job_infos.get(task.drm_jobID, dict()).get('State', 'UNK_JOB_STATE')
return {task.drm_jobID: f(task) for task in tasks}
else:
return {}
def drm_statuses(self, tasks):
job_ids = [task.drm_jobID for task in tasks]
task_infos = retry_call(
kstatus,
fargs=[job_ids],
delay=10,
tries=10,
backoff=2,
max_delay=60,
logger=tasks[0].workflow.log,
)
if len(job_ids) > 1:
task_infos = task_infos["items"]
else:
task_infos = [task_infos]
task_infos = {task_info["metadata"]["labels"]["job-name"]: task_info for task_info in task_infos}
return task_infos
def filter_is_done(self, tasks):
"""
Yield a dictionary of Slurm job metadata for each task that has completed.
"""
# jobid can be none if submission fialed
job_ids = [t.drm_jobID for t in tasks if t.drm_jobID is not None]
if job_ids:
job_infos = retry_call(do_sacct, fargs=[job_ids],
delay=10, tries=10, backoff=2, max_delay=60,
logger=tasks[0].workflow.log)
for task in tasks:
if task.drm_jobID in job_infos:
job_info = job_infos[task.drm_jobID]
if job_info['State'] in FAILED_STATES + COMPLETED_STATES:
job_info = parse_sacct(job_infos[task.drm_jobID],
tasks[0].workflow.log)
yield task, job_info
else:
assert job_info['State'] in PENDING_STATES, 'Invalid job state: `%s` for %s drm_job_id=%s' % (job_info['State'], task, task.drm_jobID)
def drm_statuses(self, tasks):
"""
:param tasks: tasks that have been submitted to the job manager
:returns: (dict) task.drm_jobID -> drm_status
"""
job_ids = [t.drm_jobID for t in tasks if t.drm_jobID is not None]
if job_ids:
job_infos = retry_call(do_sacct,
fargs=[job_ids],
delay=10,
tries=10,
backoff=2,
max_delay=60,
logger=tasks[0].workflow.log)
def f(task):
return job_infos.get(task.drm_jobID,
dict()).get('State', 'UNK_JOB_STATE')
return {task.drm_jobID: f(task) for task in tasks}
else:
return {}
