| :mod:`airflow.providers.amazon.aws.sensors.sagemaker_base` |
| ========================================================== |
| |
| .. py:module:: airflow.providers.amazon.aws.sensors.sagemaker_base |
| |
| |
| Module Contents |
| --------------- |
| |
| .. py:class:: SageMakerBaseSensor(*, aws_conn_id: str = 'aws_default', **kwargs) |
| |
| Bases: :class:`airflow.sensors.base.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:: get_hook(self) |
| |
| Get SageMakerHook |
| |
| |
| |
| |
| .. method:: poke(self, context) |
| |
| |
| |
| |
| .. method:: non_terminal_states(self) |
| |
| Placeholder for returning states with should not terminate. |
| |
| |
| |
| |
| .. method:: failed_states(self) |
| |
| Placeholder for returning states with are considered failed. |
| |
| |
| |
| |
| .. method:: get_sagemaker_response(self) |
| |
| Placeholder for checking status of a SageMaker task. |
| |
| |
| |
| |
| .. method:: get_failed_reason_from_response(self, response: dict) |
| |
| Placeholder for extracting the reason for failure from an AWS response. |
| |
| |
| |
| |
| .. method:: state_from_response(self, response: dict) |
| |
| Placeholder for extracting the state from an AWS response. |
| |
| |
| |
| |
