`airflow.providers.amazon.aws.sensors.sagemaker_endpoint`¶

Module Contents¶

class airflow.providers.amazon.aws.sensors.sagemaker_endpoint.SageMakerEndpointSensor(*, endpoint_name, **kwargs)[source]¶

Bases: airflow.providers.amazon.aws.sensors.sagemaker_base.SageMakerBaseSensor

Asks for the state of the endpoint state until it reaches a terminal state. If it fails the sensor errors, the task fails.

Parameters: job_name (str) -- job_name of the endpoint instance to check the state of

template_fields = ['endpoint_name'][source]¶

template_ext = [][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?