| :mod:`airflow.contrib.sensors.sagemaker_base_sensor` |
| ==================================================== |
| |
| .. py:module:: airflow.contrib.sensors.sagemaker_base_sensor |
| |
| |
| Module Contents |
| --------------- |
| |
| .. py:class:: SageMakerBaseSensor(aws_conn_id='aws_default', *args, **kwargs) |
| |
| Bases: :class:`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. |
| |
| .. attribute:: ui_color |
| :annotation: = #ededed |
| |
| |
| |
| |
| .. method:: poke(self, context) |
| |
| |
| |
| |
| .. method:: non_terminal_states(self) |
| |
| |
| |
| |
| .. method:: failed_states(self) |
| |
| |
| |
| |
| .. method:: get_sagemaker_response(self) |
| |
| |
| |
| |
| .. method:: get_failed_reason_from_response(self, response) |
| |
| |
| |
| |
| .. method:: state_from_response(self, response) |
| |
| |
| |
| |
