airflow.contrib.sensors.sagemaker_base_sensor

Module Contents

class airflow.contrib.sensors.sagemaker_base_sensor.SageMakerBaseSensor(aws_conn_id='aws_default', *args, **kwargs)

Bases: airflow.sensors.base_sensor_operator.BaseSensorOperator

Contains general sensor behavior for SageMaker. Subclasses should implement get_sagemaker_response() and state_from_response() methods. Subclasses should also implement NON_TERMINAL_STATES and FAILED_STATE methods.

ui_color = #ededed

poke(self, context)

non_terminal_states(self)

failed_states(self)

get_sagemaker_response(self)

get_failed_reason_from_response(self, response)

state_from_response(self, response)