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.