airflow.contrib.sensors.sagemaker_endpoint_sensor

Module Contents

class airflow.contrib.sensors.sagemaker_endpoint_sensor.SageMakerEndpointSensor(endpoint_name, *args, **kwargs)[source]

Bases:airflow.contrib.sensors.sagemaker_base_sensor.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]