airflow.contrib.sensors.sagemaker_training_sensor

Module Contents

class airflow.contrib.sensors.sagemaker_training_sensor.SageMakerTrainingSensor(job_name, print_log=True, *args, **kwargs)

Bases: airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor

Asks for the state of the training state until it reaches a terminal state. If it fails the sensor errors, failing the task.

Parameters

• job_name (str) – name of the SageMaker training job to check the state of

• print_log (bool) – if the operator should print the cloudwatch log

template_fields = ['job_name']

template_ext = []

init_log_resource(self, hook)

non_terminal_states(self)

failed_states(self)

get_sagemaker_response(self)

get_failed_reason_from_response(self, response)

state_from_response(self, response)