airflow.providers.amazon.aws.sensors.sagemaker_base

Module Contents

class airflow.providers.amazon.aws.sensors.sagemaker_base.SageMakerBaseSensor(*, aws_conn_id: str = 'aws_default', **kwargs)[source]

Bases: 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.

ui_color = #ededed[source]
get_hook(self)[source]

Get SageMakerHook

poke(self, context)[source]
non_terminal_states(self)[source]

Placeholder for returning states with should not terminate.

failed_states(self)[source]

Placeholder for returning states with are considered failed.

get_sagemaker_response(self)[source]

Placeholder for checking status of a SageMaker task.

get_failed_reason_from_response(self, response: dict)[source]

Placeholder for extracting the reason for failure from an AWS response.

state_from_response(self, response: dict)[source]

Placeholder for extracting the state from an AWS response.

Was this entry helpful?