SageMakerAutoMLSensor

Amazon

Polls the auto ML job 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

job_nameRequiredunique name of the AutoML job to watch.

Documentation

Polls the auto ML job 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 AutoML experiment state

Was this page helpful?