airflow.contrib.sensors.sagemaker_base_sensor

Module Contents

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

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[source]
poke(self, context)[source]
non_terminal_states(self)[source]
failed_states(self)[source]
get_sagemaker_response(self)[source]
get_failed_reason_from_response(self, response)[source]
state_from_response(self, response)[source]