`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]¶

Was this entry helpful?