SageMakerEndpointSensor

Amazon

Polls the endpoint state until it reaches a terminal state. Raises an AirflowException with the failure reason if a failed state is reached.

View on GitHub

Last Updated: Jan. 19, 2023

Access Instructions

Install the Amazon provider package into your Airflow environment.

Import the module into your DAG file and instantiate it with your desired params.

Parameters

endpoint_nameRequiredName of the endpoint instance to watch.

Documentation

Polls the endpoint state until it reaches a terminal state. Raises an AirflowException with the failure reason if a failed state is reached.

See also

For more information on how to use this sensor, take a look at the guide: Wait on an Amazon SageMaker endpoint state

Was this page helpful?