dml_util.runners.batch

Implementation of a Lambda function that runs a job on AWS Batch.

Environment variables: - CPU_QUEUE: The name of the CPU job queue. - GPU_QUEUE: The name of the GPU job queue. - BATCH_TASK_ROLE_ARN: The ARN of the IAM role for Batch tasks.

Classes

BatchRunner(config, input)

class dml_util.runners.batch.BatchRunner(config, input)

Bases: LambdaRunner

Parameters:

• config (EnvConfig)

• input (InputConfig)

property client

Return the AWS Batch client.

describe_job(state)

gc(state)

Garbage collect resources used by the runner.

submit()

update(state)

Update the state and return the new state, message, and response.

The gc method is called if and only if the returned state is None.