SageMakerEndpointSensor

Amazon

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

View Source

Last Updated: May. 7, 2021

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

job_namestrjob_name of the endpoint instance to check the state of

Documentation

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

Example DAGs

Improve this module by creating an example DAG.

View Source
  1. Add an `example_dags` directory to the top-level source of the provider package with an empty `__init__.py` file.
  2. Add your DAG to this directory. Be sure to include a well-written and descriptive docstring
  3. Create a pull request against the source code. Once the package gets released, your DAG will show up on the Registry.

Was this page helpful?