airflow.contrib.sensors.sagemaker_tuning_sensor

Module Contents

class airflow.contrib.sensors.sagemaker_tuning_sensor.SageMakerTuningSensor(job_name, *args, **kwargs)

Bases: airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor

Asks for the state of the tuning state until it reaches a terminal state. The sensor will error if the job errors, throwing a AirflowException containing the failure reason.

Parameters

job_name (str) – job_name of the tuning instance to check the state of

template_fields = ['job_name']

template_ext = []

non_terminal_states(self)

failed_states(self)

get_sagemaker_response(self)

get_failed_reason_from_response(self, response)

state_from_response(self, response)