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